The opportunity
Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced Driver™—to improve access to…
What you'll do
Work with AI agents to create high performance low-level kernels in a: Domain-Specific language for a custom ML accelerator
Ensure functional correctness and numerical accuracy of kernels
Conduct performance analysis of the kernels to identify performance: bottlenecks and areas that need improvement
Develop a custom harness to recursively improve kernel generation and remove the need for human involvement
Currently enrolled in an MS or PhD program in Computer Science, Computer: Engineering, Computational Science, or a related quantitative field
Demonstrated track record of using AI agents and harnesses to improve coding performance
What they're looking for
- Working knowledge of key kernels for transformer inference including performance analysis and optimization
- Familiarity with one or more low-level ML accelerator DSLs (CUDA, Triton, Gluon, etc.)
- Solid understanding of modern inference accelerator architectures
- Fluency in Python and/or C++
