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 new

  • Mountain View, CA

Responsibilities

  • Design, implement, and scale controllable generation and test-time compute algorithms for foundation world models.
  • Develop multimodal conditioning and critic-guided flow or diffusion steering methods.
  • Investigate multimodal sensor and semantic latent conditioning and representation alignment for holistic world simulation.
  • Use learned realism critics and autoresearch workflows on large-scale TPU clusters to evaluate closed-loop realism, long-tail scenario synthesis, and multimodal consistency.
  • Write clean, modular, and well-tested JAX code for simulation and world-modeling codebases while collaborating with research teams.

Requirements

  • Currently pursuing a PhD in computer science, machine learning, robotics, or a related technical field, with expected graduation between December 2027 and August 2028 and 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-matching models, or deep reinforcement-learning and critic networks.
  • Hands-on research experience in controllable generative world or sequence modeling and inference-time scaling, or multimodal spatiotemporal representation learning and closed-loop simulation, with familiarity in the other area.
  • Preferred: first-author publications at leading machine learning, computer vision, or robotics venues.
  • Preferred: experience with multi-agent closed-loop simulation, counterfactual or adversarial traffic generation, or Sim2Real discriminator and critic stabilization.
  • Preferred: experience profiling and scaling distributed training and inference on TPUs or multi-node GPUs, including XLA compilation, vectorized batched rollouts, and memory optimization.

Benefits

  • Hybrid onsite internship position.
  • Interns are eligible for the company’s benefits programs, subject to eligibility requirements.
  • Applications are accepted on a rolling basis until the role is filled.
  • Full-time summer internship for students returning to a degree program for at least one academic term.

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

Summer Internship 2027 Data Internship

  • Agoda
  • Full time
  • Others
  • 11/07/2026
  • Bangkok, Thailand

Cooperative Internship 2027 Data Internship

  • Agoda
  • Full time
  • Others
  • 11/06/2026
  • Bangkok, Thailand

Summer Internship 2027 Back end

  • Agoda
  • Full time
  • Others
  • 11/06/2026
  • Bangkok, Thailand

Campus Undergraduate Summer Internship Program

  • American Express
  • Full time
  • Canada
  • 11/04/2026
  • Toronto

2027 Software Engineer

  • Point72
  • Full time
  • Others
  • 11/04/2026
  • Bengaluru, India