2027 Summer Intern, PhD, Planner Machine Learning
Core
Research, design, and implement novel deep learning architectures and multi-task loss formulations for Waymo's Large Selection model to support autonomous driving.
Role type
PhD Research Intern (Machine Learning)
Builds
Waymo's Large Selection model for autonomous driving
Domain
Autonomous driving, deep learning, robotics
Deliverable
production ML models
Required skills
Python, JAX, deep learning frameworks, mathematical foundation in optimization and loss formulation, large-scale ML experiment design
Preferred skills
Transformer architectures, post-training/preference alignment, sequential decision making, autonomous driving and motion planning, C++, distributed data pipelines
Technologies
Python, JAX, TPU, PyTorch
Responsibilities
Research and implement novel deep learning architectures; Train and diagnose large-scale neural networks using distributed TPU infrastructure; Evaluate models across open-loop and closed-loop benchmarks
Seniority
Intern