ML Engineer - Autonomous Driving & Equity Options

Waymo; San Francisco (CA); Hybrid; USD 175,000 - 215,000; Full time

Job summary

Waymo is seeking a Senior ML Engineer to own end-to-end ML driver tasks, building scalable methods and recipes. You will analyze real-world behavior, diagnose issues, and contribute to distributed fine-tuning to improve models without regression.

Join a team that leverages vast driving data to advance ML systems powering the Waymo Driver. The role requires 3+ years in ML/CV, proficiency in Python or C++, and experience with PyTorch or JAX.

Qualifications

• Bachelor's degree in Computer Science or equivalent deep learning experience.
• 3+ years of experience in Machine Learning and/or Computer Vision.
• Experience with Python or C++.
• Experience with ML frameworks like PyTorch or JAX.

Responsibilities

• Own tasks in the ML Driver, take responsibility for task performance, create ML methods and recipes to scale and improve tasks.
• Analyze behavior of ML systems in real-world application, identify issues and root causes, advise or develop short- and long-term solutions.
• Develop methods and recipes for distributed fine-tuning enabling multiple developers to simultaneously improve the model, develop methods and recipes to avoid regression against a production system.
• Monitor ML systems in production, develop methods for automatically detecting issues or regressions, develop AI-aided analysis and debugging tooling.
• Develop and maintain model evaluation recipes and metrics for measuring and improving performance of pre-trained and fine-tuned models

Skills

Python

C++

Machine Learning

Computer Vision

Education

Bachelor's degree in Computer Science

Tools

PyTorch

JAX

Job description

Waymo is seeking a Senior ML Engineer to own end-to-end ML driver tasks, building scalable methods and recipes. You will analyze real-world behavior, diagnose issues, and contribute to distributed fine-tuning to improve models without regression.

Join a team that leverages vast driving data to advance ML systems powering the Waymo Driver. The role requires 3+ years in ML/CV, proficiency in Python or C++, and experience with PyTorch or JAX.