Waymo

Machine Learning Engineer, Simulation Realism

Waymo
Apply
2 months ago
Mountain View, CA, USASenior
H1B sponsor

Base Salary

$213k - $263k/yr

Responsibilities

  • Develop ultra-realistic world simulation technologies for autonomous vehicles.
  • Build simulation systems using foundation models for embodied agents and generative AI.
  • Apply machine learning expertise to improve simulation realism and steerability.
  • Collaborate with Waymo’s foundational AI research team to transfer research into scalable, production-ready solutions.
  • Develop and integrate machine learning models for autonomous-driving simulation and testing.

Requirements

  • 5+ years of experience in machine learning software engineering.
  • 5+ years of experience with Python.
  • 5+ years of experience with machine learning software and frameworks including PyTorch, TensorFlow, JAX, or Flax.
  • Preferred: experience creating or developing machine learning models for robotics or self-driving vehicles.
  • Preferred: strong production C++ experience, including integrating machine learning models into production systems.
  • Preferred: experience training large-scale models on GPU or TPU clusters.

Benefits

  • Hybrid work arrangement.
  • Full-time position.
  • Eligibility for a discretionary annual bonus program.
  • Eligibility for an equity incentive plan.
  • Company benefits program, subject to eligibility requirements.

Tech Stack

Categories

Waymo

About Waymo

1,001-5,000 employees

Waymo develops the Waymo Driver, a full-stack autonomous driving system, and operates the Waymo One robotaxi service that offers paid, driverless rides in select U.S. cities. The company monetizes through ride-hailing fares and partnerships integrating its technology on multiple vehicle platforms. Founded in 2009 as Google’s self-driving project, Waymo is headquartered in Mountain View and is an Alphabet subsidiary, with over 10 million rider-only trips and 100+ million autonomous miles on public roads.

Contact me