Research Scientist — Generative World Models (Hybrid)

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

Job summary

Waymo’s AI Foundations team is seeking a world-model focused researcher in a hybrid role reporting to a Software Engineering Manager in Mountain View, CA. You will conduct applied world model research, post-train Waymo’s world model, and design experiments with large generative models.

The ideal candidate holds a Masters or PhD in deep learning, 3+ years with frameworks like JAX, TensorFlow 2, or PyTorch, and hands-on experience with diffusion/flow matching, reinforcement learning, and training

Qualifications

• Masters or PhD in deep learning (generative/world modeling) required.
• 3+ years of experience with modern deep learning frameworks (e.g., JAX, TensorFlow 2, PyTorch).
• Experience with diffusion / flow matching models and reinforcement learning.

Responsibilities

• Conduct applied world model research and development.
• Post-train Waymo’s world model.
• Design compelling experiments by training and evaluating large generative models.
• Present results to peers and leadership.

Skills

Deep learning

Reinforcement learning

Generative modeling

Diffusion / flow matching

Large foundational models

Education

Masters or PhD in deep learning

Tools

JAX

TensorFlow

PyTorch

Job description

Waymo’s AI Foundations team is seeking a world-model focused researcher in a hybrid role reporting to a Software Engineering Manager in Mountain View, CA. You will conduct applied world model research, post-train Waymo’s world model, and design experiments with large generative models.

The ideal candidate holds a Masters or PhD in deep learning, 3+ years with frameworks like JAX, TensorFlow 2, or PyTorch, and hands-on experience with diffusion/flow matching, reinforcement learning, and training