Staff ML Engineer: Scalable, Low-Latency ML for Simulations

Waymo; Mountain View (CA); On-site; USD 251,000 - 310,000; Full time

Benefits offered by this job

Annual bonus program

Equity incentive plan

Generous Company benefits program

Job summary

Waymo is seeking an experienced Machine Learning Engineer in Mountain View, California, to optimize deep learning models for autonomous driving technology. In this role, you will analyze model architectures, identify performance bottlenecks, and apply advanced techniques to improve scalability and efficiency.

The ideal candidate has a strong background in deep learning, with over 5 years of experience and a relevant degree. Competitive compensation includes a salary range of $251,000 to $310,000, bonuses, and additional benefits.

Qualifications

• 5+ years of experience with deep learning architectures, especially Transformers and optimization techniques.
• Hands-on experience with quantization, pruning, and model compression methods.
• Strong programming skills in Python and experience in software development best practices.

Responsibilities

• Analyze model architectures and identify performance bottlenecks.
• Optimize model code for TPUs and GPUs.
• Design and implement low-latency serving solutions for generative models.

Skills

Deep learning architectures

Model optimization techniques

Programming in Python

JAX and Flax proficiency

Experience with ML profiling tools

Education

MS or PhD in Computer Science, Machine Learning, or related field

Tools

JAX

TensorFlow

PyTorch

Job description

Waymo is seeking an experienced Machine Learning Engineer in Mountain View, California, to optimize deep learning models for autonomous driving technology. In this role, you will analyze model architectures, identify performance bottlenecks, and apply advanced techniques to improve scalability and efficiency.

The ideal candidate has a strong background in deep learning, with over 5 years of experience and a relevant degree. Competitive compensation includes a salary range of $251,000 to $310,000, bonuses, and additional benefits.