ML Engineer, Foundation Model InfrastructurePosted today$215K

The opportunity

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced Driver™—to improve access to…

What you'll do

  • 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

What they're looking for

  • 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)