ML Engineer, Compiler

About The Company

Waymo is a leading autonomous driving technology company dedicated to transforming mobility through innovative solutions. Established as the Google Self-Driving Car Project in 2009, Waymo has evolved into a pioneer in self-driving technology, with a mission to become the world's most trusted driver. The company focuses on developing the Waymo Driver—The World's Most Experienced Driver™—aimed at enhancing access to transportation, reducing traffic-related fatalities, and improving overall road safety. With a robust portfolio that includes powering fully autonomous ride-hail services and integrating its technology across various vehicle platforms, Waymo has driven over ten million rider-only trips. Its autonomous vehicles have accumulated more than 100 million miles on public roads across more than 15 states in the U.S., complemented by tens of billions of miles in simulation, establishing Waymo as a leader in the autonomous vehicle industry.

About The Role

We are seeking a talented and experienced Neural Network Compiler Engineer to join our dynamic 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 the understanding of our neural network inference platform and optimize its capabilities. Your expertise will be instrumental in implementing compiler support for novel features of our high-performance architecture, ensuring that our neural network inference remains at the forefront of efficiency and speed. This position offers an exciting opportunity to work on cutting-edge AI and hardware integration, contributing to the development of autonomous driving systems that are safer and more reliable.

Qualifications

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

Responsibilities

• Enhance and extend the production-grade compiler to maximize neural network performance
• Collaborate with hardware architects and model developers to understand neural network inference platforms
• Develop and implement compiler support for new features of high-performance architectures
• Optimize neural network inference workflows to improve speed and efficiency
• Contribute to the development of scalable and maintainable compiler infrastructure
• Stay updated with the latest advancements in neural network algorithms and hardware architectures
  
  

Benefits

• Competitive base salary ranging from $170,000 to $216,000 USD, commensurate with experience and location
• Participation in Waymo’s discretionary annual bonus program
• Eligibility for equity incentive plans
• Comprehensive health, dental, and vision insurance
• Generous paid time off and holiday policies
• Opportunities for professional development and career growth
• Flexible work arrangements, including remote options where applicable
  
  

Equal Opportunity

Waymo is an equal opportunity employer committed to fostering an inclusive environment. We celebrate diversity and are dedicated to creating a workplace where all employees are valued and empowered. All qualified applicants will receive consideration for employment without regard to race, color, religion, gender, gender identity or expression, sexual orientation, national origin, age, disability, or any other protected status in accordance with applicable laws.