Staff Machine Learning Compiler Engineer, Compute

$226,000 - $286,000/Yr

Waymo - New York, NY

Full-time - Senior

New York, NY

Administrative and Support Services

About the position

The Staff Machine Learning Compiler Engineer at Waymo is responsible for enhancing the performance of the company's production-grade compiler, specifically for neural network inference platforms. This role involves analyzing code performance, architecting compiler support for new features, and collaborating with model developers and hardware architects to optimize hardware and software integration for maximum efficiency. The position is critical to the Compute Team's mission of delivering a robust compute platform for autonomous vehicles.

Responsibilities

• Analyze the performance characteristics of code generated by the production grade compiler and develop engineering roadmaps for improvement.
• Architect and implement compiler support for novel features of the neural network inference platform.
• Guide model developers and hardware architects to improve the efficiency and performance of inference hardware through software/hardware co-design.

Requirements

• BS degree in Computer Science/Electrical Engineering or equivalent experience and 7+ years of industry experience OR MS degree in Computer Science/Electrical Engineering and 5+ years of industry experience OR PhD degree in Computer Science/Electrical Engineering and 3+ years of industry experience.
• 3+ years of industry and/or academic experience working on compilers for neural networks or linear algebra computation targeting parallel architectures.
• 1+ years of experience in techniques used to generate code optimized for performance on a parallel architecture.
• C++ programming skills.

Nice-to-haves

• Python programming experience.
• 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.

Benefits

• Discretionary annual bonus program
• Equity incentive plan
• Generous company benefits program