Research Scientist, AI Foundations: World Modeling

Waymo; New York (NY); Hybrid; USD 213,000 - 263,000; Full time

Job summary

Waymo is seeking a senior researcher to advance the Waymo Driver world model and generative capabilities. You will design experiments, train large models, and present findings to peers and leadership in a hybrid work setting.

The role requires an advanced degree in deep learning, 3+ years using JAX/TensorFlow 2 or PyTorch, and experience with diffusion/flow matching and RL. The position involves writing code, tests, and documentation, with collaboration across teams.

Qualifications

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

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.
• Leverage AI to write high quality code, unit tests, and documentation, and automate workflows.

Skills

JAX

TensorFlow 2

PyTorch

Education

Master's or PhD in deep learning

Job description

Waymo is seeking a senior researcher to advance the Waymo Driver world model and generative capabilities. You will design experiments, train large models, and present findings to peers and leadership in a hybrid work setting.

The role requires an advanced degree in deep learning, 3+ years using JAX/TensorFlow 2 or PyTorch, and experience with diffusion/flow matching and RL. The position involves writing code, tests, and documentation, with collaboration across teams.