Machine Learning Engineer, Training

About the Role

Waymo is seeking a Machine Learning Engineer to join our ML Infrastructure team. This team collaborates with Research and Production teams to develop models in Perception and Planning, which are essential to our autonomous driving software. We focus on providing the best solutions for the entire model development lifecycle, working closely with teams at Google to scale models and solve unique ML challenges for autonomous driving.

Responsibilities

Develop infrastructure components for distributed training, including job scheduling, resource management, data distribution, and model synchronization.
Implement automation solutions for provisioning, deployment, monitoring, and scaling of distributed training infrastructure to enhance operations and reliability.
Monitor system health, diagnose and troubleshoot issues, 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 2+ years equivalent experience.
Experience with distributed systems principles and 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.
Ability to debug complex distributed systems issues.
Experience communicating updates and resolutions to customers and other partners.

Preferred Qualifications

Practical familiarity with ML accelerator profiling tools to uncover performance bottlenecks.
Familiarity with cloud computing platforms (e.g., AWS, Azure, GCP) and experience deploying and managing distributed systems in cloud environments.
Knowledge of optimization and deep learning algorithms.

About Waymo

Waymo is an autonomous driving technology company with the 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 Driver™—to improve access to mobility while saving thousands of lives now lost to traffic crashes. The Waymo Driver powers Waymo One, a fully autonomous ride-hailing service, and can also be applied to a range of vehicle platforms and product use cases. The Waymo Driver has provided over one million rider-only trips, enabled by its experience autonomously driving tens of millions of miles on public roads and tens of billions in simulation across 13+ U.S. states.

Join us in our journey to improve mobility and save lives with cutting-edge technology.

Benefits
Extracted with AI

Discretionary annual bonus program
Equity incentive plan
Generous company benefits program

Similar jobs

Last update: 23 minutes ago

dataroots

Expert Machine Learning Engineer

Join Dataroots as an Expert Machine Learning Engineer to design and deliver AI-powered solutions, focusing on machine learning models.

16 days ago

Senior C++ Computer Vision Engineer

Join a cutting-edge AI-DeepTech startup in Berlin as a Senior C++ Computer Vision Engineer. Work on world-class on-device AI technology.

12 days ago

Senior Backend Engineer C++

Join DeepL as a Senior Backend Engineer C++ to design and maintain scalable backend services using C++ and AI technologies.

16 days ago

Software Engineer - Autonomous Driving

Join Applied Intuition as a Software Engineer in Munich to tackle autonomous driving challenges with top ADAS/AV programs.

25 days ago

LLM Backend Developer

Join Persona as a LLM Backend Developer, work remotely, and develop AI-driven backend systems for top startups.

16 days ago

Senior Backend Engineer - Java, Rust, Go

Join Together AI as a Senior Backend Engineer in Amsterdam. Work with Java, Rust, and Go to build scalable backend systems.

20 days ago

Staff Software Engineer, Fullstack, Capacity & Efficiency Engineering

Join Uber as a Staff Software Engineer in Amsterdam, focusing on fullstack development and capacity efficiency engineering.

12 days ago

Cloud Data Engineer

Seeking a Cloud Data Engineer with expertise in AWS, Python, and CI/CD for a hybrid role in Hannover. Join our dynamic team!

16 days ago

Senior Fullstack Developer for AI-Driven Mission Technologies

Seeking a Senior Fullstack Developer for AI-driven mission technologies, focusing on Java, JavaScript, Python, and C++. Remote work available.

11 days ago

Information Retrieval Algorithm Engineer

Join Huawei as an Information Retrieval Algorithm Engineer to develop cutting-edge AI technologies in Amsterdam.

16 days ago

Staff Software Engineer - Backend

Join Uber as a Staff Software Engineer - Backend, focusing on membership systems. Work with Java, Python, C++, and more in Amsterdam.

12 days ago

AI Engineer

Join BCG X as an AI Engineer in Milan, Italy. Develop AI solutions, partner with clients, and drive innovation in a dynamic environment.

16 days ago

Staff Software Engineer: Data & Infrastructure Platforms - Metrics & Alerting

Join Uber's Amsterdam team as a Staff Software Engineer focusing on Data & Infrastructure Platforms, specializing in Metrics & Alerting.

14 days ago

Senior Software Engineer III - Java

Join TomTom as a Senior Software Engineer III in Amsterdam, focusing on Java and back-end development in a hybrid work environment.

11 days ago

Software Engineer - Backend - Membership

Join Uber as a Backend Software Engineer to develop member-first experiences in a collaborative team, impacting global foundations.

16 days ago

Senior Software Engineer: Configuration Management/Deployment

Join Uber's Amsterdam team as a Senior Software Engineer focusing on configuration management and deployment. Solve infrastructure challenges at scale.

14 days ago