Senior ML Compiler Engineer, Compute
Waymo · New York, NY · 2 wk ago
Information Technology$213k–$263k/yrFull-time
Leverage agentic AI
You Will Leverage agentic AI to accelerate analysis, planning and implementation of production software improvements
Analyze Performance
- Analyze the performance characteristics of code generated by our production grade compiler, and design and implement optimizations to improve that performance
- Design and implement compiler support for novel features of our high-performance architecture
Collaborate with Hardware Architects
- Work with hardware architects to understand and influence the development of our unique neural network inference platform through hardware/software codesign
Tune Neural Networks
- Work with model developers to tune their neural networks for better inference efficiency and accuracy
Skills and Experience
- BS degree in Computer Science/Electrical Engineering or equivalent experience and 5+ years of industry experience
- OR MS degree in Computer Science/Electrical Engineering and 3+ years of industry experience
- PhD degree in Computer Science/Electrical Engineering and 1+ years of industry experience
- 3+ years experience working on compilers for parallel architectures
- 1+ years experience working with ML inference or linear algebra computation
- C++ programming skills, including leveraging agentic AI acceleration
Preferred Skills
- Python programming experience
- Experience with compilers for neural networks, especially MLIR-based
- Knowledge of computer architectures used for neural network inference, and neural network performance characteristics
- Knowledge of the principles behind popular machine learning and neural network algorithms and applications