Senior Machine Learning Engineer for Autonomous Driving

4 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 cutting-edge technology that improves access to mobility while saving thousands of lives lost to traffic crashes.

The Waymo ML Infrastructure team is responsible for developing models in Perception and Planning that are core to our autonomous driving software. We work closely with Research and Production teams to ensure our partners receive the best solutions for the entire model development lifecycle.

We develop a set of libraries and tools that enhance TensorFlow and JAX, addressing scalability, reliability, and performance challenges faced by Waymo's ML practitioners. Our team is passionate about increasing ML accelerator efficiency, fine-tuning multimodal LLMs for autonomous driving tasks, and validating newly trained DNNs when deployed into the full onboard software stack.

Key Responsibilities:
  • Develop infrastructure components necessary for distributed training
  • 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.

Salary Range: $192,000 - $243,000 USD



  • Mountain View, Arkansas, United States Waymo Full time

    Waymo: A Leader in Autonomous Driving TechnologyWaymo is a pioneering company in the field of autonomous driving technology, with a mission to create the most trusted driver. With a rich history dating back to the Google Self-Driving Car Project in 2009, Waymo has made significant strides in building the Waymo Driver, the world's most experienced driver....


  • 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

    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 SummaryWaymo is an autonomous driving technology company with a mission to be the most trusted driver. We are seeking a highly skilled Machine Learning Engineer to join our team and contribute to the development of our autonomous driving technology.About the RoleThe successful candidate will have a strong background in machine learning and software...


  • Mountain View, Arkansas, United States Wayve Full time

    About the RoleAt Wayve, we're seeking a highly skilled Senior Software Engineer, Autonomous Systems to join our team. As a key member of our software engineering organization, you will be responsible for ensuring the performance and efficiency of our autonomous systems, working closely with embedded and distributed infrastructure, machine learning, and...


  • 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

    At Waymo, we're working towards a future where autonomous driving technology improves access to mobility while saving thousands of lives. Our mission is to be the most trusted driver, and we're looking for talented engineers to help us achieve this goal.The Waymo ML Infrastructure team is responsible for developing models in Perception and Planning that are...


  • Mountain View, Arkansas, United States Waymo Full time

    Waymo Driver Software DeveloperWaymo is an autonomous driving technology company with a mission to be the most trusted driver. Our team focuses on 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...


  • Mountain View, Arkansas, United States Waymo Full time

    Job SummaryWaymo is an autonomous driving technology company with a mission to be the most trusted driver. The company has focused on 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,...


  • Mountain View, Arkansas, United States Waymo Full time

    Join Waymo's Team of Experts in Autonomous DrivingWaymo is a pioneering company in the field of autonomous driving technology, with a mission to create the most trusted driver. Our team has made significant progress in building the Waymo Driver, a cutting-edge technology that has provided over one million rider-only trips and has driven tens of millions of...


  • Mountain View, Arkansas, United States Waymo Full time

    Waymo Driver DeveloperWaymo is an autonomous driving technology company with a mission to be the most trusted driver. Our team focuses on 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...


  • 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

    Waymo is a leader in autonomous driving technology, with a mission to improve access to mobility while saving thousands of lives lost to traffic crashes. Our software allows the Waymo Driver to perceive the world around it, make the right decision for every situation, and deliver people safely to their destinations.As a software engineer or researcher at...


  • Mountain View, Arkansas, United States Waymo Full time

    Waymo's MissionWaymo is an autonomous driving technology company with a mission to be the most trusted driver. Our team focuses on 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 DriverThe Waymo Driver powers Waymo One, a fully autonomous...


  • Mountain View, Arkansas, United States Waymo Full time

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


  • Mountain View, Arkansas, United States Waymo Full time

    Waymo's Compute Team is tasked with a critical mission: delivering the compute platform responsible for running the autonomous vehicle's software stack. To achieve this, we architect and create high-performance custom silicon, develop system-level compute architectures, and collaborate with teammates to optimize hardware and software for maximum...


  • Mountain View, Arkansas, United States Waymo Full time

    Job Title: Software Engineer, SensorsWaymo is a leader in autonomous driving technology, 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 lives.About the RoleWe are seeking a talented Software Engineer to join our Sensor Systems team....


  • Mountain View, Arkansas, United States Waymo Full time

    Waymo is a pioneering autonomous driving technology company dedicated to making mobility safer and more accessible. Our mission is to be the most trusted driver, and we're committed to achieving this goal through innovative technology and collaboration.The Waymo Compute Team plays a critical role in this mission, responsible for delivering the compute...


  • Mountain View, Arkansas, United States Orby AI Full time

    Join Orby AI's Mission to Revolutionize AutomationOrby AI is building the next-generation Automation AI platform to automate repetitive processes. Our platform provides a unique "observe, learn, and automate" end-to-end experience powered by an actions-based foundation model. We observe a user's activities, identify repetitive work steps, and generate...


  • Mountain View, Arkansas, United States Waymo Full time

    Job SummaryWaymo is a leader in autonomous driving technology, and we're seeking a skilled Senior Software Engineer to join our TaaS (Transportation as a Service) and Ride-Hailing Infrastructure team. As a key member of our team, you will design and develop platform and infrastructure that supports various ride-hailing businesses, providing APIs for both our...