CronJobs

ai-ml jobs

2027 Summer Intern, PhD, Planner Machine Learning

Waymo · San Francisco, California

hybridjunior$176,800–$176,800Posted Sep 29, 2026PythonJAXPyTorchTPUTransformersDeep Learning

Apply on the employer site

About this role

**Waymo — 2027 Summer Intern (PhD) — Planner, Machine Learning** Waymo is an autonomous driving technology company with the mission to be the world’s most trusted driver. The Waymo Driver powers Waymo’s fully autonomous ride-hail service and can be applied across vehicle platforms and product use cases. **Software Engineering** builds the brains of Waymo’s fully autonomous driving technology—helping the Waymo Driver perceive the world, make the right decisions, and deliver people safely. --- ## What you will do - Research, design, and implement novel deep learning architectures, multi-task loss formulations, and preference/ranking objectives in **Python** and **JAX** for Waymo’s **Large Selection** model. - Train and diagnose large-scale neural networks using **distributed TPU** infrastructure, analyzing training dynamics, gradient conditioning, and representation quality. - Evaluate trained models across **open-loop** and **closed-loop** benchmarks, triaging real-world driving scenarios to connect modeling choices with physical vehicle behavior. ## What you have - Currently pursuing a **Ph.D. in Machine Learning** or a related quantitative field. - Strong hands-on proficiency in **Python** with at least one modern deep learning framework (e.g., **JAX** or **PyTorch**). - Solid mathematical foundation in deep learning, optimization, loss formulation, and empirical model diagnostics. - Experience designing, running, and analyzing rigorous ML experiments on large-scale datasets. ## We prefer - First-author publications at top-tier ML, robotics, or vision venues (e.g., **NeurIPS, ICML, ICLR, CoRL, ICRA, CVPR**, etc.). - Experience with **transformer architectures**, **post-training/preference alignment**, or **sequential decision making**. - Familiarity with **autonomous driving** and **motion planning**. - Experience with **C++** and/or navigating large-scale production ML codebases and distributed data pipelines. --- **Note:** This is a **hybrid onsite** internship position. Resumes will be accepted on a **rolling basis** until the role is filled. To be considered for multiple roles, you must apply to each one individually—please apply to the **top 3 roles** you’re interested in. **Pay (Hourly PhD Pay): $85–$85 USD**

Listing freshness

CronJobs last confirmed this listing 1h ago. If its source stops confirming the opening for seven days, this page is removed from active inventory.

Browse all software engineering jobs →

Follow fresh jobs in Discord