Waymo is looking to improve the efficiency and performance of its ML models on specialized hardware for autonomous driving, both in vehicles and at scale for simulation.
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.
- Strong track record of high quality research in machine learning, computer systems, and/or computer architecture.
- 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.
