Machine Learning Engineer, Simulation Evaluation

1 month, 1 week ago
Full-time
Senior
Software Development

Waymo

Waymo is an autonomous driving technology company building the Waymo Driver and operating Waymo One, its fully autonomous ride-hailing service.

Autonomous vehicles, robotics, AI, ride-hailing / mobility tech
Founded 2009
$21600M raised

Description

  • Lead the design, development, and deployment of evaluation methods for assessing the realism of multimodal world models and generative simulation systems.
  • Architect and implement scalable machine learning pipelines for tuning, evaluating, and deploying large-scale discriminator models.
  • Evaluate video generation techniques using realism metrics such as temporal stability, multimodal consistency, geometric discrepancy, and condition following.
  • Apply vision-language models to assess semantic understanding and controllability in world simulation products.
  • Collaborate with research teams across Waymo and Alphabet to bring 4D world modeling and generative AI advances into production systems.
  • Mentor engineers on the team and provide technical guidance on architecture and execution.

Requirements

  • Bachelor's, Master's, or PhD in computer science, machine learning, robotics, or a related field.
  • 5+ years of experience in machine learning engineering or applied deep learning, with a portfolio of shipped products or peer-reviewed publications.
  • Proficiency in Python and hands-on experience with machine learning frameworks such as JAX, Flax, or PyTorch.
  • Experience designing and implementing evaluation frameworks for complex systems or machine learning models.
  • Track record of training large-scale generative models, such as diffusion models, flow matching models, or vision-language models (preferred).
  • PhD with demonstrated success delivering machine learning products focused on 3D generative models, world models, or video generation (preferred).
  • Experience simulating sensor data, including camera, lidar, and radar, or modeling semantic scenes (preferred).
  • Experience developing autonomous systems, robotics software, or autonomous vehicle simulations (preferred).
  • Experience training and optimizing large-scale models on GPU or TPU clusters for efficient production serving (preferred).
  • Professional experience writing C++ for high-performance production systems (preferred).

Benefits

  • Base salary range of $213,000 to $263,000 USD.
  • Eligibility for Waymo’s discretionary annual bonus program.
  • Eligibility for Waymo’s equity incentive plan.
  • Access to Waymo’s generous company benefits program, subject to eligibility requirements.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

ML Infrastructure Engineer

x.ai 51-250 Internet Software & Services

SpaceXAI is hiring an ML Infrastructure Engineer to build and optimize the machine learning platform that powers recommendations on X.

Ansible C++ Linux Puppet Python PyTorch Rust
14 hours, 59 minutes ago

Senior Machine Learning Engineer, Safety

Reddit 1K-5K Internet Software & Services

Reddit is hiring a remote-friendly Machine Learning Engineer to build and improve safety systems that support enforcement of Reddit rules using large language models.

Deep Learning LLM Machine Learning NLP Python PyTorch TensorFlow
15 hours, 29 minutes ago

Senior, Machine Learning Engineer - 3D Perception

Torc 251-1K Road & Rail

Torc is hiring a Senior Machine Learning Engineer – 3D Perception to develop and deploy production perception models for autonomous trucks and improve Bird's Eye View understanding across its autonomy stack.

C++ Computer Vision Deep Learning Machine Learning Python PyTorch
15 hours, 44 minutes ago

Applied AI Engineer

Unframe Inc. 51-200 Technology, Information and Internet

Unframe is hiring an Applied AI Engineer to work with enterprise customers and internal teams on deploying AI-driven solutions that connect business problems to production systems.

Machine Learning Python React TypeScript
1 day, 14 hours ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers