2027 Summer Intern, PhD, Machine Learning - Generative World Models & Controllable Simulation

Waymo; 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 is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced Driver™—to improve access to mobility while saving thousands of lives now lost to traffic crashes. The Waymo Driver powers Waymo’s fully autonomous ride-hail service and can also be applied to a range of vehicle platforms and product use cases. The Waymo Driver has provided over ten million rider-only trips, enabled by its experience autonomously driving over 100 million miles on public roads and tens of billions in simulation across 15+ U.S. states.

Software Engineering builds the brains of Waymo's fully autonomous driving technology. Our software allows the Waymo Driver to perceive the world around it, make the right decision for every situation, and deliver people safely to their destinations. We think deeply and solve complex technical challenges in areas like robotics, perception, decision-making and deep learning, while collaborating with hardware and systems engineers. If you’re a software engineer or researcher who’s curious and passionate about Level 4 autonomous driving, we'd like to meet you.

Waymo interns partner with leaders in the industry on projects that create impact to the company. We believe learning is a two-way street: applying your knowledge while providing you with opportunities to expand your skill‑set. Interns are an important part of our culture and our recruiting pipeline. Join us at Waymo for a fun and rewarding internship!

You will:

• Design, implement, and scale controllable generation and test-time compute algorithms, multimodal conditioning, and critic-guided flow/diffusion steering—for foundation world models.
• Investigate multimodal sensor + semantic latent conditioning, and representation alignment to advance toward holistic world simulation.
• Leverage learned realism critics and agentic autoresearch workflows on large-scale TPU clusters to evaluate closed-loop interactive realism, long-tail scenario synthesis, and multimodal consistency.
• Write clean, modular, and well-tested JAX code in core simulation and world-modeling codebases, collaborating closely with researchers across Sim Realism, Foundation Models, and Closed-Loop Training teams.

You have:

• Currently pursuing a PhD in Computer Science, Machine Learning, Robotics, or a related technical field, with an anticipated graduation date between December 2027 and August 2028 (returning to a degree program for at least one academic term after the internship).
• Core Implementation Fluency (Mandatory): Strong tensor implementation and debugging skills in Python and JAX/Flax or PyTorch, with hands‑on experience training and modifying modern neural architectures (such as Transformers, Diffusion/Flow-Matching models, or deep RL/critic networks).
• Deep Research Expertise in Either: (a) Controllable Generative World / Sequence Modeling & Inference-Time Scaling (hands‑on research experience in continuous flow matching / Rectified Flows, score-based diffusion guidance/steering, test‑time compute / SMC, or parameter‑efficient finetuning) OR (b) Multimodal Spatiotemporal Representation Learning & Closed-Loop Simulation (hands‑on research experience in 4D/video tokenization, latent dynamics alignment such as REPA/DINO alignment, inverse RL / adversarial discriminators, or multi‑agent trajectory modeling), with working familiarity in the other.

We prefer:

• First‑author publications at top‑tier machine learning, computer vision, or robotics venues (e.g., NeurIPS, ICML, ICLR, CVPR, ECCV, CoRL, RSS, ICRA, or equivalent venues) in generative world models, flow/diffusion steering, controllable generation, or autonomous driving simulation.
• Experience with multi‑agent closed‑loop simulation, counterfactual/adversarial traffic generation, or Sim2Real discriminator/critic stabilization (e.g., Waymax, WOMD, NuPlan, MGAIL, or reward‑hacking mitigation).
• Experience profiling and scaling distributed training and inference workloads on TPUs or multi‑node GPUs (such as XLA compilation, vectorized batched rollouts, and memory optimization) and interest in external conference publication on Waymax/WOMD.

Note: This will be a hybrid onsite internship position.

The expected hourly rate for this full-time position is listed below. Interns are also eligible to participate in the Company’s generous benefits programs, subject to eligibility requirements.

Hourly PhD Pay

$85—$85 USD