ML Engineer: Driver Behavior Refinement & Validation

Waymo LLC; Mountain View (CA); Hybrid; USD 175,000 - 215,000; Full time

Benefits offered by this job

Annual bonus program

Equity incentives

Company benefits

Job summary

Waymo LLC in Mountain View, CA, is hiring for a hybrid ML role focused on the Waymo Driver. You will own tasks in ML Driver, analyze real-world behavior, and develop scalable methods and recipes to improve models. The role requires a CS degree and 3+ years of ML/CV experience, with Python or C++ and PyTorch/JAX proficiency.

We offer a base salary and bonus/equity programs in a supportive team environment. This full-time hybrid position reports to a Technical Lead Manager and involves

Qualifications

• Bachelor's in CS or equivalent deep learning experience.
• 3+ years of ML/Computer Vision experience.
• 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

PyTorch

JAX

Education

Bachelor's degree in Computer Science or equivalent

MS or PhD in ML/Robotics/CS

Tools

PyTorch

JAX

Job description

Waymo LLC in Mountain View, CA, is hiring for a hybrid ML role focused on the Waymo Driver. You will own tasks in ML Driver, analyze real-world behavior, and develop scalable methods and recipes to improve models. The role requires a CS degree and 3+ years of ML/CV experience, with Python or C++ and PyTorch/JAX proficiency.

We offer a base salary and bonus/equity programs in a supportive team environment. This full-time hybrid position reports to a Technical Lead Manager and involves