Senior Machine Learning Compiler Engineer (Compute)

Join Waymo's Compute Team as a Senior Machine Learning Compiler Engineer. In this role, you will be responsible for analyzing and optimizing the performance of code generated by our production-grade compiler, designing and implementing compiler support for novel features of our high-performance architecture, and collaborating with hardware architects and model developers. This position offers a hybrid work schedule and a comprehensive benefits package.

• Analyzing the performance characteristics of code generated by the production-grade compiler and designing and implementing optimizations to improve that performance.
• Designing and implementing compiler support for novel features of the high-performance architecture, and working with hardware architects to influence the development of the neural network inference platform.
• Collaborating with model developers to tune their neural networks for better inference efficiency and accuracy.
• MS degree in Computer Science/Electrical Engineering and 3+ years of industry experience
• C++ programming skills, including leveraging agentic AI acceleration
• BS degree in Computer Science/Electrical Engineering or equivalent experience and 5+ years of industry experience OR
• PhD degree in Computer Science/Electrical Engineering and 1+ years of industry experience
• 1+ years experience working with ML inference or linear algebra computation
• 3+ years experience working on compilers for parallel architectures
• 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