Staff ML Lead: Fine-Tune & Release Billion-Parameter Models

Waymo; San Francisco (CA); Hybrid; USD 238,000 - 302,000; Full time

Benefits offered by this job

Discretionary annual bonus

Equity incentive plan

Generous company benefits

Job summary

A leading autonomous driving technology company in San Francisco is seeking a Senior Machine Learning Engineer to lead the fine-tuning of advanced ML models. You will drive the technical roadmap while managing a high-performing team. The ideal candidate has over 5 years of experience in ML and Computer Vision, alongside strong Python and ML frameworks (PyTorch, JAX) skills. This position offers a competitive salary along with benefits, and the opportunity to work in a hybrid role.

Qualifications

• 5+ years experience in Machine Learning and/or Computer Vision.
• 2+ years experience in managing ML engineering or research teams.
• Experience with ML frameworks.

Responsibilities

• Lead the fine-tuning of multi-billion parameter models.
• Implement ML methods for distilling expert models into multi-task models.
• Build high-performing teams through development and hiring.

Skills

Machine Learning

Computer Vision

Python

Education

Bachelor's degree in Computer Science or similar

Tools

PyTorch

JAX

C++

Job description

A leading autonomous driving technology company in San Francisco is seeking a Senior Machine Learning Engineer to lead the fine-tuning of advanced ML models. You will drive the technical roadmap while managing a high-performing team. The ideal candidate has over 5 years of experience in ML and Computer Vision, alongside strong Python and ML frameworks (PyTorch, JAX) skills. This position offers a competitive salary along with benefits, and the opportunity to work in a hybrid role.