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
Senior Machine Learning Performance Engineer
Join Wayve as a Senior Machine Learning Performance Engineer to optimize large-scale training jobs and improve GPU efficiency.
Software Engineer, Commercialization Engineering
Join Waymo as a Software Engineer in Commercialization Engineering, focusing on infrastructure for driverless cars.
Principal Software Engineer, Storage & Compute
Lead software engineer role focusing on storage & compute infrastructure for autonomous driving tech at Waymo.
ML Compiler Engineer, Compute
Join Waymo as an ML Compiler Engineer in NYC, enhancing neural network performance with cutting-edge technology.
Software Engineer, Reasoning Foundation
Join Waymo as a Software Engineer in Reasoning Foundation to develop cutting-edge autonomous driving technology.
Expert Machine Learning Engineer
Join Dataroots as an Expert Machine Learning Engineer to design and deliver AI-powered solutions, focusing on machine learning models.
Machine Learning Engineer - LLM Fine-tuning and Performance
Join NVIDIA as a Machine Learning Engineer specializing in LLM fine-tuning and performance optimization. Work with cutting-edge ML technologies.
Software Engineer AI/ML, Devices and Services
Join Google as a Software Engineer AI/ML to develop systems for devices like Pixel and Nest, enhancing supply chain processes.
Machine Learning Engineer, Cloud AI
Join Qualcomm as a Machine Learning Engineer to develop AI solutions for mobile, edge, auto, and IoT products.
Software Engineer III, Machine Learning
Join Google as a Software Engineer III in Machine Learning at YouTube, focusing on large-scale systems and AI technologies.
Senior Software Engineer - LLM
Join Bonfy.AI as a Senior Software Engineer to develop and optimize scalable machine learning models using Python, TensorFlow, and cloud platforms.
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.
Machine Learning Engineer
Join MoonPay as a Machine Learning Engineer to build and maintain ML infrastructure, collaborating with data scientists and cross-functional teams.
AI Engineer - Machine Learning and Robotics
Join Blueprint as an AI Engineer in Machine Learning and Robotics, focusing on scalable AI model training systems. Hybrid role in Redmond, WA.
Software Engineer - Autonomous Driving
Join Applied Intuition as a Software Engineer in Munich to tackle autonomous driving challenges with top ADAS/AV programs.
Machine Learning Engineer
Join Refuel as a Machine Learning Engineer to develop core ML algorithms, improve datasets, and collaborate on product scalability.
Software Engineer III, Machine Learning
Join Google as a Software Engineer III in Machine Learning, focusing on large-scale systems and AI.
Senior Backend Engineer C++
Join DeepL as a Senior Backend Engineer C++ to design and maintain scalable backend services using C++ and AI technologies.
Software Engineer, Machine Learning Infrastructure
Join Tesla as a Software Engineer in ML Infrastructure to optimize and scale neural network training with Python, C++, and PyTorch.
Machine Learning Engineer
Join Ema Unlimited as a Machine Learning Engineer in SF Bay Area, working on cutting-edge AI solutions with a focus on NLP and ML technologies.
Machine Learning Engineer
Join Snap Inc. as a Machine Learning Engineer in New York, NY. Develop AI models, collaborate with teams, and drive innovation.
LLM Backend Developer
Join Persona as a LLM Backend Developer, work remotely, and develop AI-driven backend systems for top startups.
Machine Learning Engineer
Join Uber as a Machine Learning Engineer to enhance Uber Eats search experience using ML, data analysis, and engineering.
Machine Learning Software Engineer
Join Arm's Machine Learning Software team in Galway to develop high-performance ML applications on Arm hardware.
