GrepJob
Waymo

ML Engineer, Foundation Model Infrastructure

Waymo
Apply
about 4 hours ago
Kirkland, WA, USA +3 moreMid Level / Senior
H1B Sponsor

Base Salary

$175k - $215k/yr

Responsibilities

  • Build and operate petabyte-scale data systems and ML pipelines.
  • Shepherd foundation models from research prototypes to robust components.
  • Create automated infrastructure for benchmarking and monitoring models.
  • Process massive datasets and train/deploy complex models using frameworks like Flume and JAX.
  • Drive improvements in the speed, reliability, and efficiency of the ML development lifecycle.
  • Collaborate with AI Foundations, ML, and Platform experts for on-road improvements.

Requirements

  • Master's degree in Computer Science, Machine Learning, Robotics, or a similar field.
  • Proficiency in Python and C++.
  • Familiarity with modern deep learning frameworks like Pytorch, JAX, or Tensorflow.
  • Experience building or maintaining large-scale data pipelines or ML infrastructure.

Benefits

  • Eligible for Waymo’s discretionary annual bonus program.
  • Participation in equity incentive plan.
  • Generous Company benefits program.

Tech Stack

Categories

AI & MLData EngineeringDevOps