ML Engineer, Foundation Model Infrastructure

We are seeking a highly skilled ML Engineer to join our team and help build the Waymo Driver—The World's Most Experienced DriverTM—by developing machine learning solutions addressing open problems in autonomous driving. The successful candidate will be responsible for building and operating petabyte-scale data systems and ML pipelines, shepherding cutting-edge foundation models from research prototypes to robust components, and partnering with AI Foundations, ML, and Platform experts to transform model innovations into tangible on-road improvements.

Requirements

• Masters degree in Computer Science, Machine Learning, Robotics, similar technical field of study, or equivalent practical experience
• Proficiency in Python
• Proficiency in C++
• Familiarity with one of the modern deep learning frameworks (e.g. Pytorch, JAX, Tensorflow)
• Experience building or maintaining large-scale data pipelines or ML infrastructure (e.g., Flume, Spark, Borg, Kubeflow)

Benefits

• Discretionary annual bonus program
• Equity incentive plan
• Generous Company benefits program