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 related field
  • Proficiency in Python and C++
  • Familiarity with deep learning frameworks like Pytorch, JAX, or Tensorflow
  • Experience in large-scale data pipelines or ML infrastructure
  • Hands-on software engineering skills, preferably with complex codebases

Responsibilities

  • Build and operate petabyte-scale data systems and ML pipelines
  • Shepherd foundation models from prototypes to components within Waymo Driver
  • Create infrastructure for benchmarking and monitoring ML models
  • Utilize large-scale compute frameworks to process data and train models
  • Drive efficiency in the ML development lifecycle
  • Collaborate with experts to translate model innovations into practical applications

Benefits

  • Eligible for discretionary annual bonus program
  • Participation in equity incentive plan
  • Generous company benefits package including health and wellness programs
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: