ML Engineer, Foundation Model Infrastructure

Waymo is an autonomous driving technology company focused on improving mobility and safety through its advanced driver technology. The ML Engineer will build and operate data systems and ML pipelines, develop foundation models, and collaborate with various teams to enhance the performance of Waymo's autonomous vehicles.

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