ML Engineer, Compiler

About The Company

Waymo is a pioneering autonomous driving technology company committed to transforming mobility and enhancing safety on the roads. Established in 2009 as the Google Self-Driving Car Project, Waymo has evolved into a leader in self-driving technology with a clear mission to become the world's most trusted driver. The company focuses on developing the Waymo Driver, an advanced autonomous driving system that powers its fully autonomous ride-hail service and is adaptable to various vehicle platforms and use cases. With a track record of over ten million rider-only trips and more than 100 million miles driven on public roads across more than 15 states, Waymo continues to innovate and improve the safety and accessibility of autonomous transportation. The company's relentless pursuit of technological excellence aims to save lives and provide equitable mobility solutions worldwide.

About The Role

We are seeking a talented and dedicated Neural Network Compiler Engineer to join our team. In this role, you will be responsible for maximizing the performance of our neural networks by enhancing and extending our production-grade compiler. You will collaborate closely with hardware architects and model developers to deepen our understanding of our neural network inference platform and develop support for novel features of our high-performance architecture. Your work will directly impact the efficiency and effectiveness of Waymo’s autonomous driving systems, enabling safer and more reliable autonomous vehicles. The ideal candidate will possess strong technical expertise in compiler development, parallel computing, and machine learning inference, along with a passion for advancing autonomous vehicle technology.

Qualifications

• BS degree in Computer Science, Electrical Engineering, or a related field with 3+ years of industry experience OR
• MS degree in Computer Science, Electrical Engineering, or a related field with 1+ years of industry experience OR
• PhD degree in Computer Science, Electrical Engineering, or equivalent experience
• At least 1 year of industry and/or academic experience working with compilers and parallel computing
• Experience with ML inference or linear algebra computations in industry or academia
• Proficiency in C++ programming
  
  

Responsibilities

• Enhance and extend our production-grade neural network compiler to improve performance and support new features
• Collaborate with hardware architects and model developers to understand and optimize neural network inference platforms
• Implement compiler support for novel features of high-performance architecture to maximize efficiency
• Analyze and optimize neural network models for inference performance and resource utilization
• Contribute to the development of tools and workflows for neural network compilation and deployment
• Stay current with advancements in compiler technologies, neural network architectures, and machine learning algorithms
  
  

Benefits

• Competitive salary range of $170,000—$216,000 USD, commensurate with experience and location
• Participation in Waymo’s discretionary annual bonus program and equity incentive plan
• Comprehensive benefits package including health, dental, and vision insurance
• Generous paid time off and flexible work arrangements
• Opportunities for professional growth and development within a pioneering autonomous vehicle company
• A collaborative and innovative work environment focused on safety and technological excellence
  
  

Equal Opportunity

Waymo is an equal opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all employees. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, gender identity, sexual orientation, national origin, age, disability, or any other protected characteristic. We believe that a diverse workforce fosters innovation and strengthens our mission to build safer, smarter transportation solutions.