Waymo is looking to improve the efficiency and performance of its ML models for autonomous driving through compiler optimization and novel programming interfaces.
Requirements
- Experience in ML compilers and MLIR.
- Strong programming skills in C++.
- Experience with hardware acceleration, GPU and/or TPU optimizations.
- Experience with JAX, Pallas and Triton.
Responsibilities
- Develop innovative compiler optimization techniques to enhance the efficiency of Waymo ML models.
- Prototype novel programming interfaces for authoring highly efficient and adaptable ML algorithms and kernels.
- Implement solutions using C++ and Python.
Other
- Progressing towards PhD in Computer Science, Electrical Engineering, Machine Learning, or related technical fields.
- 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.
