6 hours ago
Mountain View, CA, USAIntern
H1B sponsor
Responsibilities
- Create high-performance low-level kernels in a domain-specific language for a custom ML accelerator using AI agents.
- Ensure kernel functional correctness and numerical accuracy.
- Analyze kernel performance to identify bottlenecks and improvement opportunities.
- Develop a custom harness to recursively improve kernel generation and reduce human involvement.
Requirements
- Currently enrolled in an MS or PhD program in Computer Science, Computer Engineering, Computational Science, or a related quantitative field.
- Demonstrated experience using AI agents and harnesses to improve coding performance.
- Working knowledge of transformer-inference kernels, including performance analysis and optimization.
- Familiarity with one or more low-level ML accelerator DSLs, such as CUDA, Triton, or Gluon.
- Preferred: understanding of modern inference accelerator architectures.
- Preferred: fluency in Python and/or C++.
- Preferred: knowledge of ML numerical formats and their accuracy-versus-performance tradeoffs.
Benefits
- Eligible interns may participate in Waymo’s company benefits programs, subject to eligibility requirements.
- Hybrid onsite internship position.
- Applications are accepted on a rolling basis until the role is filled.
Categories
About Waymo
Waymo develops the Waymo Driver, a full-stack autonomous driving system, and operates the Waymo One robotaxi service that offers paid, driverless rides in select U.S. cities. The company monetizes through ride-hailing fares and partnerships integrating its technology on multiple vehicle platforms. Founded in 2009 as Google’s self-driving project, Waymo is headquartered in Mountain View and is an Alphabet subsidiary, with over 10 million rider-only trips and 100+ million autonomous miles on public roads.