Machine Learning Engineer, Model Optimization

Waymo

Mountain View, California, United States
Base: $170,000 - $216,000 usd; bonus/equity: discr...
**
Optimize flops utilization
Model architecture/hardware co-development
Sparse representation optimization
** Waymo is seeking a Machine Learning Engineer focused on model optimization to enhance the performance of its autonomous driving technology. The ideal candidate should have a background in machine learning or computer vision, with experience in optimizing model architectures and inference for real-world applications. **

Job Summary

  • The Perception team builds the system which learns the spatial-temporal representation and their semantic meanings of the surrounding environment of the autonomously driving vehicle (ADV).
  • You will optimize FLOPs utilization in model training and model inference through model architecture/ hardware co-development, optimize for a naturally sparse representation.
  • Waymo employees are also eligible to participate in Waymo’s discretionary annual bonus program, equity incentive plan, and generous Company benefits program, subject to eligibility requirements.

Matching Summary

Match Score: 75

** Waymo is seeking a Machine Learning Engineer focused on model optimization to enhance the performance of its autonomous driving technology. The ideal candidate should have a background in machine learning or computer vision, with experience in optimizing model architectures and inference for real-world applications. **

Salary

Base: $170,000 - $216,000 USD; Bonus/Equity: discretionary annual bonus program, equity incentive plan; Benefits: generous Company benefits program

Skills & Requirements

Must-have

  • Optimize FLOPs utilization
  • Model architecture/hardware co-development
  • Sparse representation optimization
  • Real-time inference optimization
  • Machine Learning
  • Computer Vision
  • Python
  • ML frameworks (PyTorch/JAX)

Nice-to-have

  • Research collaboration
  • Large-scale data learning
  • Complex model architectures
  • Onboard system criticality

Key Requirements

  • Bachelors in Computer Science or equivalent experience
  • 3+ years experience in ML/Computer Vision
  • Experience with Python
  • Experience with ML frameworks

Work Rights

Not specified

Tailored Resume

Cover Letter