Drimva

Waymo

Research Scientist, World Model Post-Training

$213K–$263K

Mountain View, CA, USA +5 · Full-time · Posted Sep 22, 2026

Location
Mountain View, CA, USA
San Francisco, CA, USA
New York, NY, USA
Mountain View, California, United States
New York, New York, United States
San Francisco, California, United States
Experience
3+ years of experience with modern deep learning frameworks.
Education
Master’s or PhD in deep learning in generative / world modeling.

About the job

Conduct applied world model research and post-train Waymo’s world model by designing experiments to train and evaluate large generative models.

What you'll do

  • Conduct applied world model research and development and post-train Waymo’s world model.
  • Design experiments to train and evaluate large generative models.
  • Present results to peers and leadership; use AI to write code, unit tests, and documentation and automate workflows.

What they ask for

  • Master’s or PhD in deep learning in generative / world modeling.
  • 3+ years of experience with modern deep learning frameworks.
  • Experience with diffusion / flow matching models.
  • Experience with reinforcement learning and training large foundational models.

Nice to have

  • Experience with post-training of foundational models and reinforcement learning for diffusion / flow matching models.
  • Experience with distillation or autonomous vehicles.

Pay

$213,000 — $263,000 USD (base salary)

Tools

JAX, TensorFlow 2, Pytorch

AI focus

Applied world model research and post-training; training and evaluating large generative models, with reinforcement learning and diffusion / flow matching.

Model-written summary · the original posting is authoritative

First seen Oct 10, 2026 · Last verified Nov 5, 2026 (UTC)