Staff Machine Learning Engineer, Simulation

About the Company

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. The Waymo Driver powers Waymo’s fully autonomous ride-hail service and has provided over ten million rider-only trips.

Responsibilities

• Lead the development of cutting edge Deep Learning and machine learning models to enhance human-led triaging and introduce automation for high-volume workflows
• Proactively monitor and assimilate best practices to develop a Reinforcement Learning from human preference-based data collection and evaluation system
• Enhance User Feedback Analysis and collaborate with product teams to design tools for multi-label classifications, sentiment assessment, and trend analysis
• Oversee the production and optimization of machine learning models to assess Waymo’s expansive fleet of vehicles
• Drive technical direction and provide guidance to the team
• Work closely with PMs and TPMs to define product requirements and align technical agenda with business objectives
• Collaborate with multiple teams including Prediction, Planning, and Research to deliver on strategic efforts

Requirements

• B.S. in Computer Science, Robotics, Machine Learning, or similar technical field
• Strong coding experience in C++ and/or Python
• Experience in Foundational Models, VLM, or Deep Learning
• 5+ years of hands-on experience in machine learning projects
• Experience with ML frameworks such as TensorFlow, PyTorch, or Hugging Face transformers
• Expertise in deep learning models and ML deployment at scale

Preferred Qualifications

• M.S. or Ph.D. in Computer Science or a related quantitative field
• Deep learning experience with Transformers
• Experience building tools for applied machine learning, including MLOps and evaluation techniques
• Large-scale data processing and analytical skills