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.
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.
Machine Learning Engineer
Join Refuel as a Machine Learning Engineer to develop core ML algorithms, improve datasets, and collaborate on product scalability.
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.
Software Engineer III, Machine Learning
Join Google as a Software Engineer III in Machine Learning, focusing on large-scale systems and AI.
Machine Learning Engineer
Join Uber as a Machine Learning Engineer to enhance Uber Eats search experience using ML, data analysis, and engineering.
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.
Machine Learning Software Engineer
Join Arm's Machine Learning Software team in Galway to develop high-performance ML applications on Arm hardware.
Remote Software Engineer - Machine Learning and Cloud Infrastructure
Join Helm.ai as a Remote Software Engineer to develop ML tools, build cloud infrastructure, and work on AI technology.
Senior Backend Software Engineer
Senior Backend Software Engineer at Waymo in Warsaw. Develop critical systems for autonomous driving technology.
AI Engineer for Software-Defined Vehicles
AI Engineer role focusing on software-defined vehicles, requiring skills in AI, ML, and cloud technologies.
Machine Learning Engineer
Join Snap Inc. as a Machine Learning Engineer in Los Angeles. Develop and deploy ML models to enhance user experience. Competitive salary and benefits.
Principal Engineer, Commercialization
Lead Principal Engineer role in Commercialization at Waymo, focusing on backend systems and technical leadership.
Next-Gen Deep Learning Engineer
Join Zenseact as a Deep Learning Engineer in Munich. Work on cutting-edge AI for autonomous driving.
Senior Software Engineer, ML Infrastructure
Senior Software Engineer for ML Infrastructure at Nuro, focusing on optimizing ML development and performance.
Machine Learning Engineer - Ads
Join as a Machine Learning Engineer focusing on Ads, developing predictive models in a hybrid role in New York.
Machine Learning Engineer
Join Snap Inc. as a Machine Learning Engineer in New York, NY. Develop AI models, collaborate with teams, and drive innovation.
