Senior Machine Learning Engineer

2 weeks ago


Mountain View, Arkansas, United States Waymo Full time
Job Title: Senior Machine Learning Engineer, Training

Waymo 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 Role

We are seeking a Senior Machine Learning Engineer to join our Hybrid team, reporting to our TLM of Machine Learning Training. As a key member of our team, you will be responsible for developing the infrastructure components necessary for distributed training, implementing automation solutions for provisioning, deployment, monitoring, and scaling of distributed training infrastructure, and monitoring system health, diagnosing, and performing routine maintenance tasks to ensure the reliability of the distributed training infrastructure.

Responsibilities
  • Develop and maintain distributed training infrastructure components
  • Implement automation solutions for provisioning, deployment, monitoring, and scaling of distributed training infrastructure
  • Monitor system health, diagnose, and perform routine maintenance tasks to ensure the reliability of the distributed training infrastructure
  • Identify performance bottlenecks and optimization opportunities
  • Improve the developer experience and performance of our scalable ML framework
Requirements
  • Bachelor's degree in Computer Science, Engineering, or related field, or 4+ years equivalent experience
  • Experience building distributed systems for production environments
  • Solid Python or C++ skills
  • Prior experience with Machine Learning frameworks (e.g., TensorFlow, PyTorch) and distributed training algorithms
Preferred Qualifications
  • Practical familiarity using ML accelerator profiling tools to uncover performance bottlenecks
  • Experience deploying and managing distributed systems in cloud environments
  • Knowledge of optimization and deep learning algorithms

Waymo employees are eligible to participate in our 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 $192,000 - $243,000 USD.



  • 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 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 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

    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, 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: 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 cutting-edge technology that improves access to mobility while saving thousands of lives lost to traffic crashes.The Waymo ML Infrastructure team...


  • 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 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 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

    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

    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 Waymo Full time

    Join Waymo's Compute TeamWaymo is a leader in autonomous driving technology, with a mission to improve access to mobility while saving thousands of lives. Our Compute Team is responsible for delivering the compute platform that powers our autonomous vehicles. We're seeking a talented Machine Learning Compiler Engineer to join our team.Key...


  • 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 key player in the industry, we're committed to building the most advanced driverless technology. Our team is dedicated to developing cutting-edge solutions that enable safe and efficient transportation.Job SummaryWe're...


  • 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 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 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 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

    About the RoleWaymo is a pioneering autonomous driving technology company dedicated to creating a safer and more accessible transportation system. As a key member of our team, you will play a crucial role in developing and refining the Waymo Driver, our cutting-edge autonomous driving system.Key ResponsibilitiesCollaborate with a talented team of engineers...


  • 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...