ML Engineer, Foundation Model Infrastructure

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. The role focuses on developing machine learning solutions for autonomous driving, building and operating large-scale data systems and ML pipelines, and collaborating with research teams to improve the Waymo Driver's capabilities.

Responsibilities:

• Build and operate the petabyte-scale data systems and ML pipelines at the heart of Waymo's foundation model development
• Shepherd cutting-edge foundation models from research prototypes to robust components within the Waymo Driver
• Create the automated infrastructure for rigorously benchmarking, continuously monitoring, and safely releasing models
• Wield large-scale compute and frameworks like Flume and JAX to process massive datasets and train/deploy complex models
• Drive significant leaps in the speed, reliability, and efficiency of the end-to-end ML development lifecycle
• Partner with AI Foundations, ML, and Platform experts to transform model innovations into tangible on-road improvements