Staff Machine Learning Engineer for Model Optimization

2 weeks ago


Mountain View, Arkansas, United States Waymo Full time
Job Title: Staff Machine Learning Engineer for Model Optimization

Waymo is a leading autonomous driving technology company with a mission to be the most trusted driver. Our team is dedicated to building the Waymo Driver, the world's most experienced driver, to improve access to mobility while saving thousands of lives lost to traffic crashes.

The Waymo Driver powers Waymo One, a fully autonomous ride-hailing service, and can be applied to a range of vehicle platforms and product use cases. With over one million rider-only trips and tens of millions of miles driven on public roads, our team is committed to making machine learning more accessible to teams at Waymo.

Job Summary

We are seeking a staff engineer with model optimization expertise to help us improve compute performance on our car. In this hybrid role, you will report to our Head of Model Optimization and collaborate with world-class Waymo ML scientists in perception, planner, research, and simulation.

Key Responsibilities
  • Lead the collaboration with Waymo ML scientists to identify opportunities in systems and models to make ML workloads faster.
  • Lead projects from proposals through execution by leading and mentoring junior engineers.
  • Apply model optimization, efficient deep learning techniques, and ML software improvements.
  • Collaborate on foundation models and ML systems with external partners such as CoreML, Google Brain, and Deepmind.
Requirements
  • M.S. in CS, EE, Deep Learning, or a related field.
  • 2+ years of experience as a technical lead.
  • 3+ years of experience on model optimization or efficient deep learning techniques.
  • Strong Python or C++ programming skills.
  • Solid experience with designing, training, and debugging deep learning models to achieve the highest scores/accuracies.
Preferred Qualifications
  • PhD in CS, EE, Deep Learning, or a related field.
  • Deep knowledge on system performance, GPU optimization, or ML compiler.
  • 5+ years of experience on model optimization or efficient deep learning techniques.

Waymo employees are eligible to participate in Waymo's discretionary annual bonus program, equity incentive plan, and generous Company benefits program, subject to eligibility requirements. The expected base salary range for this full-time position across US locations is $226,000 - $286,000 USD.



  • Mountain View, Arkansas, United States Waymo Full time

    About the RoleWaymo is a pioneering autonomous driving technology company with a mission to revolutionize mobility and save lives. As a Staff Machine Learning Engineer, you will play a critical role in optimizing our car's compute performance, collaborating with world-class ML scientists, and leading projects from proposal to execution.Key...


  • Mountain View, Arkansas, United States Waymo Full time

    About the RoleWaymo is a leading autonomous driving technology company with a mission to create the most trusted driver. As a Staff Machine Learning Engineer, you will play a critical role in improving the performance of our car's machine learning models.Key ResponsibilitiesCollaborate with world-class Waymo ML scientists to identify opportunities for...


  • Mountain View, Arkansas, United States Wayve Full time

    About WayveWayve is a leading developer of Embodied AI technology, committed to creating a diverse and inclusive culture that values everyone's unique skills and perspectives. Our advanced AI software and foundation models enable vehicles to perceive, understand, and navigate complex environments, enhancing the usability and safety of automated driving...


  • Mountain View, Arkansas, United States Wayve Full time

    About WayveWayve is a leading developer of Embodied AI technology, pioneering advanced AI software and foundation models that enable vehicles to perceive, understand, and navigate complex environments. Our mission is to create autonomy that propels the world forward.The RoleWe are seeking a skilled Machine Learning Performance Engineer to join our Machine...


  • Mountain View, Arkansas, United States Waymo Full time

    Job Title: Senior Machine Learning Engineer, Runtime OptimizationWaymo is a leading autonomous driving technology company with a mission to be the most trusted driver. Our team is dedicated to building the Waymo Driver, a world-class autonomous driving system that improves access to mobility while saving thousands of lives lost to traffic crashes.The ML...


  • Mountain View, Arkansas, United States Waymo Full time

    Job Title: Staff Machine Learning Engineer, RuntimeWaymo is a leading autonomous driving technology company with a mission to be the most trusted driver. Our team is responsible for building the Waymo Driver, the world's most experienced driver, to improve access to mobility while saving thousands of lives lost to traffic crashes.The ML Platform team at...


  • Mountain View, Arkansas, United States Waymo Full time

    Job Title: Staff Machine Learning Engineer, RuntimeWaymo is a leading autonomous driving technology company with a mission to be the most trusted driver. Our team is responsible for building the Waymo Driver, a cutting-edge technology that powers our fully autonomous ride-hailing service and can be applied to various vehicle platforms and product use...


  • Mountain View, Arkansas, United States Waymo Full time

    Join the Driver Understanding and Evaluation Team at WaymoWaymo is a leader in autonomous driving technology, with a mission to improve access to mobility while saving thousands of lives. Our Driver Understanding and Evaluation team is responsible for developing rich metrics for understanding the behavior of the Waymo Driver in the real world.Your RoleIn...


  • Mountain View, Arkansas, United States Waymo Full time

    About the RoleWaymo is a pioneering autonomous driving technology company with a mission to become the most trusted driver. Since its inception as the Google Self-Driving Car Project in 2009, Waymo has focused on developing the Waymo Driver - The World's Most Experienced DriverTM - to improve access to mobility while saving thousands of lives lost to traffic...


  • Mountain View, Arkansas, United States Wayve Full time

    About WayveWayve is a pioneering company in the field of Embodied AI technology, dedicated to creating a safer and more efficient future for autonomous driving. Our mission is to propel the world forward by developing intelligent, mapless, and hardware-agnostic AI products that enhance the usability and safety of automated driving systems.The RoleWe are...


  • Mountain View, Arkansas, United States Waymo Full time

    Join Waymo's ML Platform TeamWaymo is a leading autonomous driving technology company with a mission to improve access to mobility while saving thousands of lives. Our team is dedicated to building the Waymo Driver, the world's most experienced driver, to power our fully autonomous ride-hailing service and various vehicle platforms.About the RoleWe are...


  • Mountain View, Arkansas, United States Wayve Full time

    About WayveWayve is a pioneering developer of Embodied AI technology, dedicated to creating a safer and more efficient future for autonomous driving. Our cutting-edge AI software and foundation models enable vehicles to navigate complex environments, enhancing the usability and safety of automated driving systems.The RoleWe're seeking an experienced Machine...


  • Mountain View, Arkansas, United States Waymo Full time

    Job Title: Machine Learning Engineer, TrainingWaymo is a leading autonomous driving technology company with a mission to be the most trusted driver. Our team is dedicated to building the Waymo Driver, a software stack that enables our vehicles to navigate complex environments and interact with their surroundings. We are seeking a highly skilled Machine...


  • Mountain View, Arkansas, United States Waymo Full time

    Job Title: Senior Machine Learning Engineer, TrainingWaymo is a leading autonomous driving technology company with a mission to create the most trusted driver. Our team is dedicated to building the Waymo Driver, a software that improves access to mobility while saving thousands of lives lost to traffic crashes.The Waymo ML Infrastructure team works closely...


  • Mountain View, Arkansas, United States Waymo Full time

    Job Title: Senior Machine Learning Compiler EngineerWaymo is a leading autonomous driving technology company with a mission to create the most trusted driver. Our team is responsible for delivering the compute platform that powers our autonomous vehicles, and we're seeking a highly skilled Senior Machine Learning Compiler Engineer to join our ranks.About the...


  • Mountain View, Arkansas, United States Waymo Full time

    Join Waymo's Compute TeamWaymo is a leader in autonomous driving technology, with a mission to create the most trusted driver. Our Compute Team is responsible for delivering the compute platform that powers our autonomous vehicles. We're seeking a talented Senior Machine Learning Compiler Engineer to join our team.About the RoleWe're looking for a skilled...


  • Mountain View, Arkansas, United States Waymo Full time

    Job Title: Senior Machine Learning Engineer, TrainingWaymo is a leading autonomous driving technology company with a mission to create the most trusted driver. Our team is dedicated to building the Waymo Driver, a software that improves access to mobility while saving thousands of lives lost to traffic crashes.About the RoleWe are seeking a Senior Machine...


  • Mountain View, Arkansas, United States Wayve Full time

    About WayveWayve is a pioneering company in the field of embodied intelligence, developing cutting-edge AI technology to navigate complex environments. Our innovative approach integrates the latest advancements in machine learning, enabling us to understand and adapt to diverse urban landscapes like never before.Our MissionWe're dedicated to solving one of...


  • Mountain View, Arkansas, United States Waymo Full time

    Job Title: Machine Learning Compiler EngineerWaymo is a leading autonomous driving technology company with a mission to create the most trusted driver. Our team is responsible for delivering the compute platform that powers our autonomous vehicles, and we're seeking a talented Machine Learning Compiler Engineer to join our ranks.About the RoleWe're looking...


  • Mountain View, Arkansas, United States Waymo Full time

    About the RoleWaymo is a leading autonomous driving technology company with a mission to be the most trusted driver. Our team is dedicated to building the Waymo Driver, a cutting-edge technology that improves access to mobility while saving thousands of lives lost to traffic crashes.Job SummaryWe are seeking a highly skilled Machine Learning Engineer,...