Your browser does not support javascript! Please enable it, otherwise web will not work for you.

Senior Machine Learning Engineer (Infra)

Home > Python programming jobs

Senior Machine Learning Engineer (Infra) in USA

  • Mountain View, CA

Responsibilities

  • Build scalable systems for training and fine-tuning large-scale models that evaluate driving behaviors.
  • Design and scale distributed systems for dataset generation, model training, and evaluation across the ML lifecycle.
  • Contribute to production and optimization of machine learning models assessing Waymo’s vehicle fleet.
  • Provide guidance on architecture and technical direction while owning large, complex systems.
  • Collaborate cross-functionally to translate product and business goals into measurable technical deliverables.

Requirements

  • M.S. or Ph.D. in Computer Science, Machine Learning, Artificial Intelligence, or a related technical field, or equivalent practical experience.
  • At least 5 years of experience developing, designing, scaling, training, deploying, and optimizing large-scale machine learning systems.
  • Experience contributing to machine learning tooling and frameworks such as PyTorch, JAX, TensorFlow, Ray, or similar.
  • Strong expertise in distributed training techniques, including gradient sharding and optimization strategies for scaling large models.
  • Experience with ML accelerator profiling tools to identify performance bottlenecks.
  • Preferred: 7+ years of machine learning infrastructure experience.
  • Preferred: experience in autonomous vehicles, robotics, or complex simulation environments.
  • Preferred: familiarity with large-scale simulation platforms and their integration with ML training workflows.

Benefits

  • Eligibility for a discretionary annual bonus program, equity incentive plan, and company benefits program, subject to eligibility requirements.
  • Full-time position with location-specific salary ranges determined by work location, experience, training, education, and skill level.

Salary: $213k - $263k/yr

Waymo

On the journey to be the world's most trusted driver. With the Waymo Driver, we can improve mobility while saving thousands of lives. Download the Waymo One app and ride today. Waymo reaches out to candidates from official channels only (e.g. directly from @waymo.com email addresses, or through o...

Similar positions

Research Scientist - Machine Learning

  • ISEE
  • Full time
  • USA
  • 09/13/2026
  • Remote

Software Engineer in Test

  • Roadie
  • Full time
  • USA
  • 09/13/2026
  • Salary: Competitive
  • Remote

Principal Software Engineer - Builder Experience

  • Elastic
  • Full time
  • Remote
  • 09/13/2026
  • Salary: €73k-€116k
  • Greece

Principal Software Engineer - Builder Experience

  • Elastic
  • Full time
  • Remote
  • 09/13/2026
  • Salary: zł 369k-zł 584k
  • Poland

Senior Devops Engineer

  • Woliba
  • Full time
  • USA
  • 09/13/2026
  • Remote

Listings

by


Job Type

Language

Country

CodingJobBoard.com © 2026