Your browser does not support javascript! Please enable it, otherwise web will not work for you.

2027 Summer Intern, PhD

Home > Python programming jobs

2027 Summer Intern, PhD in USA

  • Mountain View, CA

Responsibilities

  • Design, implement, and scale reinforcement learning post-training and distillation algorithms for foundation world models in simulation and autonomous driving.
  • Develop learned reward models, realism critics, and discriminators in a closed-loop multi-agent simulator.
  • Run closed-loop ablation studies on large-scale TPU clusters using autonomous-driving behavior metrics.
  • Investigate simulator exploitation and distribution shift through agentic auto-research workflows and fleet-scale analysis.
  • Write clean, modular, and well-tested JAX code for simulation and autonomous-driving codebases.
  • Collaborate with researchers and engineers across simulation, world modeling, and autonomous-driving teams.

Requirements

  • Currently pursuing a PhD in Computer Science, Machine Learning, Robotics, or a related technical field.
  • Expected PhD graduation between December 2027 and August 2028, with at least one academic term remaining after the internship.
  • Strong tensor implementation and debugging skills in Python and JAX/Flax or PyTorch.
  • Hands-on experience training and modifying modern neural architectures such as Transformers, diffusion or flow models, or deep reinforcement learning networks.
  • Research experience in reinforcement learning and post-training or generative world, sequence, or trajectory modeling, with working familiarity in the other area.
  • Preferred: first-author publications at top-tier machine learning or robotics venues such as NeurIPS, ICML, ICLR, CVPR, CoRL, RSS, or ICRA.
  • Preferred: experience with multi-agent closed-loop simulation, Sim2Real transfer, reward-hacking mitigation, or tools and environments such as Waymax, WOMD, NuPlan, IsaacGym, or MuJoCo.
  • Preferred: experience profiling and scaling distributed training on TPUs or multi-node GPUs, including XLA compilation, vectorized batched rollouts, and memory optimization.
  • Preferred: familiarity with C++/Python interfaces.

Benefits

  • Hybrid onsite internship position.
  • Full-time internship with an expected hourly PhD pay rate of $85 per hour.
  • Interns are eligible for the company’s benefits programs, subject to eligibility requirements.
  • Resumes are accepted on a rolling basis until the role is filled.

Waymo

On the journey to be the world's most trusted driver. With the Waymo Driver, we can improve mobility while saving thousands of lives. Download the Waymo One app and ride today. Waymo reaches out to candidates from official channels only (e.g. directly from @waymo.com email addresses, or through o...

Similar positions

2027 Software Engineer

  • Point72
  • Full time
  • Others
  • 10/25/2026
  • Bengaluru, India

2027 Summer Intern, MS/PhD

  • Waymo
  • Full time
  • USA
  • 10/25/2026
  • Salary: Competitive
  • San Francisco, CA

Campus Graduate Masters Summer Internship

  • American Express
  • Full time
  • Canada
  • 10/25/2026
  • Salary: Competitive
  • Toronto

2027 Summer Intern, MS/PhD

  • Waymo
  • Full time
  • USA
  • 10/24/2026
  • Salary: Competitive
  • Mountain View, CA

Winter 2027 Business Operations Engineer

  • Anduril Industries
  • Full time
  • USA
  • 10/21/2026
  • Costa Mesa, CA