ML Compiler Engineer, Compute
Waymo
about 1 month ago
Bengaluru, India
Mid Level / Senior
H1B Sponsor
Responsibilities
- Maximize performance of neural networks by enhancing and extending the production-grade compiler.
- Collaborate with hardware architects and model developers to understand the neural network inference platform.
- Implement compiler support for novel features of the high-performance architecture.
Requirements
- BS degree in Computer Science/Electrical Engineering or equivalent practical experience and 3+ years of industry experience, or MS degree with 1+ years, or PhD.
- 1+ years of industry and/or academic experience with compilers and parallel computing.
- 1+ years of industry and/or academic experience working with ML inference or linear algebra computations.
- C++ programming skills.
Benefits
- Eligible for Waymo’s discretionary annual bonus program.
- Participation in equity incentive plan.
- Generous Company benefits program.
Tech Stack
C++Python
Categories
AI & MLData Engineering