ML Engineer, Foundation Model Infrastructure

Description

You will: 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 You have: Masters degree in Computer Science, Machine Learning, Robotics, similar technical field of study, or equivalent practical experience Proficiency in Python Proficiency in C++ Familiari