Hybrid PhD Intern: Controllable Generative World Models

Waymo LLC; Mountain View (CA); Hybrid; USD 97,000 - 138,000; Full time

Benefits offered by this job

Hybrid onsite internship

Generous benefits

Internship program

Job summary

Waymo LLC is seeking a PhD candidate for a hybrid onsite internship in Mountain View to advance autonomous driving simulations and world-modeling research. You will design and implement scalable algorithms in Python/JAX/PyTorch, contribute to multimodal world models, and collaborate with researchers across teams.

This internship offers a rewarding experience, generous benefits, and the possibility to publish in top venues. The hourly rate is $85 USD, with eligibility for Waymo benefits.

Qualifications

• Pursuing a PhD in CS, ML, robotics with expected graduation between Dec 2027 and Aug 2028.
• Strong Python experience with JAX/Flax or PyTorch.
• Hands-on research in controllable generative world models, diffusion/flow-based methods, or multimodal representation learning.

Responsibilities

• Design, implement, and scale controllable generation and test-time compute algorithms for foundation world models.
• Investigate multimodal sensor conditioning and representation alignment for world simulation.
• Leverage realism critics and autoresearch workflows on TPU clusters to evaluate realism and multimodal consistency.
• Write clean, modular, and well-tested Python code in core simulation and world-modeling codebases, collaborating with researchers.

Skills

Python

JAX/Flax

PyTorch

Transformer models

Diffusion/Flow-Matching

Education

PhD in Computer Science, ML, Robotics

Tools

TPUs

GPUs

JAX

Job description

Waymo LLC is seeking a PhD candidate for a hybrid onsite internship in Mountain View to advance autonomous driving simulations and world-modeling research. You will design and implement scalable algorithms in Python/JAX/PyTorch, contribute to multimodal world models, and collaborate with researchers across teams.

This internship offers a rewarding experience, generous benefits, and the possibility to publish in top venues. The hourly rate is $85 USD, with eligibility for Waymo benefits.