2026 Summer Intern, MS/PhD, ML Training Infra and Efficiency

Waymo interns partner with leaders in the industry on projects that create impact to the company. We believe learning is a two-way street: applying your knowledge while providing you with opportunities to expand your skill-set. Interns are an important part of our culture and our recruiting pipeline. Join us at Waymo for a fun and rewarding internship!

You will:

• Evaluate end-to-end ML training workflows to pinpoint feature gaps and construct infrastructure components that streamline training processes.
• Improve the developer experience and performance of our scalable ML framework
• Apply model optimization, efficient deep learning techniques and ML software improvements to Waymo's ML systems.

You have:

• Python or C++ experience
• Scale model training to multiple accelerators
• Machine Learning frameworks (e.g., TensorFlow, JAX, or PyTorch) and distributed training algorithms
• Practical familiarity using ML accelerator profiling tools to uncover performance bottlenecks

We prefer:

• ML compiler optimization such as TVM or XLA
• Understand ML compiler infrastructure, such as of HLO and MLIR
• Reinforcement learning

Note: This will be a hybrid onsite internship position. We will accept resumes on a rolling basis until the role is filled. To be in consideration for multiple roles, you will need to apply to each one individually - please apply to the top 3 roles you are interested in.