PhD ML & Robotics Research Intern (Hybrid)

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

Benefits offered by this job

Benefits programs

Hybrid onsite internship

Job summary

Waymo LLC in Mountain View is seeking a PhD candidate for a research internship in reinforcement learning and autonomous driving. You will design, implement, and scale post-training algorithms and run large-scale experiments on TPU clusters with researchers across teams.

The role emphasizes clean JAX/PyTorch code, collaboration with simulation and world modeling groups, and a strong foundation in Python and ML theory.

Qualifications

• You are pursuing a PhD with anticipated graduation between Dec 2027 and Aug 2028.
• You have strong Python and ML framework experience (JAX/Flax or PyTorch).
• You have deep RL or generative world model expertise (or both).

Responsibilities

• Design, implement, and scale post-training RL algorithms for foundation models.
• Develop learned reward models and discriminators in a closed-loop simulator.
• Run large-scale ablation studies on TPU/multi-node GPUs.
• Write clean, modular JAX/PyTorch code across simulation and world modeling teams.

Skills

Python

JAX/Flax

PyTorch

Transformers

Diffusion/Flow models

Deep RL

Tensor debugging

Education

PhD candidate in Computer Science / ML / Robotics

Tools

TPU clusters

multi-node GPUs

XLA

C++/Python interfaces

Job description

Waymo LLC in Mountain View is seeking a PhD candidate for a research internship in reinforcement learning and autonomous driving. You will design, implement, and scale post-training algorithms and run large-scale experiments on TPU clusters with researchers across teams.

The role emphasizes clean JAX/PyTorch code, collaboration with simulation and world modeling groups, and a strong foundation in Python and ML theory.