ML Engineer, Realistic Autonomous Simulation

Waymo LLC; Mountain View (CA); Hybrid; USD 213,000 - 263,000; Full time

Benefits offered by this job

Annual bonus program

Equity incentive plan

Benefits program

Job summary

Waymo LLC in Mountain View, California, is seeking a senior ML/AI engineer for the Simulator Team who will help advance ultra-realistic world simulation for autonomous vehicles. This hybrid role reports to a Staff Technical Lead.

You will design and train multimodal world models, scale foundation models, and drive reinforcement learning pipelines, collaborating with Waymo's AI research team to productionize solutions for real-world driving scenarios.

Qualifications

• 5+ years of experience with ML software engineering.
• 5+ years of Python experience.
• 5+ years of ML software experience with machine learning frameworks including PyTorch, TensorFlow, or JAX/Flax.
• Demonstrated experience building, training, or scaling world models or multi-modal models on large-scale datasets (e.g., video, images, text)
• Hands‑on experience with closed‑loop training, policy learning, or reinforcement learning frameworks

Responsibilities

• Drive innovation in ultra-realistic world simulation for autonomous vehicles by developing and scaling foundation models for embodied agents.
• Design and train multimodal world models and generative architectures using diverse sensory data (such as RGB images, video, and text).
• Drive closed-loop training and reinforcement learning pipelines to advance simulation realism, policy optimization, and decision‑making for autonomous driving.
• Work with Waymo's foundational AI research team to transfer research into scalable, production‑ready solutions.

Skills

ML software engineering

Python

ML frameworks (PyTorch, TensorFlow, JX

World models / multi-modal models

Reinforcement learning

Tools

C++ production

GPU/TPU clusters

Job description

Waymo LLC in Mountain View, California, is seeking a senior ML/AI engineer for the Simulator Team who will help advance ultra-realistic world simulation for autonomous vehicles. This hybrid role reports to a Staff Technical Lead.

You will design and train multimodal world models, scale foundation models, and drive reinforcement learning pipelines, collaborating with Waymo's AI research team to productionize solutions for real-world driving scenarios.