2026 Summer Intern, MS/PhD, ML Training Infra… at Waymo · Mountain…

Summary: Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. The role involves evaluating ML training workflows, improving developer experience, and applying optimization techniques to enhance Waymo's ML systems. Responsibilities: - 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 Required Qualifications: - 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 Preferred Qualifications: - ML compiler optimization such as TVM or XLA - Understand ML compiler infrastructure, such as of HLO and MLIR - Reinforcement learning Required Skills: Python, C++, Machine Learning frameworks, TensorFlow, JAX, PyTorch, Distributed training algorithms, ML accelerator profiling tools, ML compiler optimization, TVM, XLA, ML compiler infrastructure, HLO, MLIR, Reinforcement learning Internship Start Date: Start in 2026 Summer Benefits: Participate in the Company’s generous benefits programs, subject to eligibility requirements