Waymo is looking to improve the performance and efficiency of its autonomous driving technology, specifically the Waymo Driver, to enhance mobility and safety on public roads.
Requirements
- Python or C++ experience
- Scale model training to multiple accelerators
- Reinforcement learning
- Machine Learning frameworks (e.g., TensorFlow, JAX, or PyTorch) and distributed training algorithms
- Practical familiarity using ML accelerator profiling tools to uncover performance bottlenecks
- ML compiler optimization such as TVM or XLA
- Understand ML compiler infrastructure, such as of HLO and MLIR
Responsibilities
- Identify model performance bottlenecks and optimization opportunities
- 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
Other
- 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
- Interns are also eligible to participate in the Company’s generous benefits programs, subject to eligibility requirements
- This will be a hybrid onsite internship position
- We will accept resumes on a rolling basis until the role is filled
