Waymo

ML Engineer, Foundation Model Infrastructure

Waymo$175K — $215K *
Transportation
Less than 5 years of experience
Job Overview by Ladders

Qualifications

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

Responsibilities

  • Build and operate petabyte-scale data systems and ML pipelines for model development
  • Shepherd foundation models from prototypes to robust components
  • Create infrastructure for benchmarking, monitoring, and releasing models
  • Process massive datasets using large-scale compute and frameworks like Flume and JAX
  • Drive speed, reliability, and efficiency in the ML development lifecycle
  • Collaborate with experts to turn model innovations into real-world applications

Benefits

  • Eligibility for discretionary annual bonus program
  • Participation in equity incentive plan
  • Generous company benefits program
Full Job Description
In this hybrid role, you will report to a Senior Research Scientist.

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

We prefer:
  • Strong hands-on SWE skills, able to drive development of large, complex shared codebases
  • Experience in AV planning and related research
  • Experience designing and building distributed systems or MLOps platforms (e.g., model versioning, experiment tracking, CI/CD for ML)
  • Prior work in an industrial or research setting developing methodologies for the evaluation of ML models


The expected base salary range for this full-time position across US locations is listed below. Actual starting pay will be based on job-related factors, including exact work location, experience, relevant training and education, and skill level. Your recruiter can share more about the specific salary range for the role location or, if the role can be performed remote, the specific salary range for your preferred location, during the hiring process.

Waymo employees are also eligible to participate in Waymo's discretionary annual bonus program, equity incentive plan, and generous Company benefits program, subject to eligibility requirements.

Salary Range

$175,000-$215,000 USD

About Waymo

Waymo LLC is a self-driving technology development company. It is a subsidiary of Alphabet Inc., the parent company of Google. Waymo develops autonomous driving technology and provides ride-hailing services through its Waymo One program. The company has been testing its self-driving technology on public roads since 2009 and has logged millions of miles of autonomous driving. Waymo has partnerships with several automakers and has been working on developing autonomous trucks for use in logistics.
Learn more about Waymo
Size
2,000 employees
Industry
Founded
2009

Similar Jobs

More Jobs at Waymo

More Transportation Jobs

Find similar ML Engineer, Foundation Model Infrastructure jobs: