PhD ML Intern: Generative World Models & Simulation

First Roles; Mountain View (CA); Híbrido; USD 97.000 - 138.000; Jornada completa

Ventajas ofrecidas por este puesto de trabajo

Hybrid onsite internship

Competitive hourly pay

Waymo benefits (eligibility)

Descripción de la vacante

Waymo is seeking a PhD candidate for a Summer Intern role in Mountain View, CA, focused on generative world models and controllable simulation. You will design and test scalable, multimodal models, collaborating with researchers across domains and delivering robust JAX/Python implementations.

Hybrid onsite arrangement supported; expected hourly compensation reflects PhD pay standards. The role emphasizes deep research expertise in controllable generative modeling, multimodal learning, and

Formación

• Strong Python, JAX/PyTorch experience for ML research.
• Experience with diffusion/flow models or RL is a plus.
• Publications in top ML/robotics venues desired.
• PhD candidate with anticipated graduation 2027–2028.

Responsabilidades

• Design and implement controllable generative world models.
• Experiment with multimodal conditioning for simulation.
• Collaborate with research teams and scale experiments on TPU clusters.
• Write clean, well-tested code in JAX/Python.

Conocimientos

Python

JAX/Flax

PyTorch

TensorFlow

Educación

PhD in Computer Science

Machine Learning

Herramientas

TPU

GPUs

XLA

JAX

Descripción del empleo

Waymo is seeking a PhD candidate for a Summer Intern role in Mountain View, CA, focused on generative world models and controllable simulation. You will design and test scalable, multimodal models, collaborating with researchers across domains and delivering robust JAX/Python implementations.

Hybrid onsite arrangement supported; expected hourly compensation reflects PhD pay standards. The role emphasizes deep research expertise in controllable generative modeling, multimodal learning, and