Foundation Model Data, Software Engineer

Job Summary

The Waymo AI Foundations team is seeking a skilled Software Engineer to build and operate the petabyte-scale data systems and ML pipelines at the heart of Waymo's foundation model development. This role will shepherd cutting-edge foundation models from research prototypes to robust components within the Waymo Driver, develop automated, high-performance infrastructure, and drive significant leaps in the speed, reliability, and efficiency of the end-to-end ML development lifecycle.

Responsibilities

• Build and operate petabyte-scale data systems and ML pipelines
• Develop automated, high-performance infrastructure for data solutions
• Process massive datasets using frameworks like Flume/Beam and JAX
• Collaborate with AI Foundations, ML, and Platform experts to transform model innovations
• Drive significant improvements to Waymo’s product through model-centric enhancements

Requirements

• Undergraduate degree in Computer Science, Machine Learning, Robotics, or a related technical field
• Proficiency in Python and C++
• Experience building or maintaining large-scale data pipelines or ML infrastructure
• Strong hands-on software engineering skills
• Experience with distributed systems or MLOps platforms
• Familiarity with modern deep learning frameworks
• Experience in AV planning and related research
• Prior data curation and multimodal experience

Qualifications

• Experience in an industrial or research setting developing methodologies for data-centric improvement of ML models
• Previous data curation experience and multimodal experience

Benefits

Benefits for this role include:

• Health, dental, vision, life, disability insurance
• Roth 401(k) with company match
• Paid Time Off: 20 days per year, accruing at a rate of 6.15 hours per pay period for the first five years of employment
• Sick Time: 40 hours/year (statutory, where applicable); 5 days/event (discretionary)
• Maternity Leave (Short-Term Disability + Baby Bonding): 28-30 weeks
• Baby Bonding Leave: 18 weeks
• Holidays: 13 paid days per year