Staff Machine Learning Swe Infra

Waymo

Mountain View, CA, United States
Base: $238,000 - $302,000 usd; bonus/equity: discr...
On-site
Large-scale ml model architectures
Scalable systems for training
Data engineering and model development
Waymo is seeking a Staff Machine Learning Software Engineer to lead the development of scalable machine learning systems for their autonomous vehicle technology. The ideal candidate will have extensive experience in machine learning infrastructure, with a strong focus on model training, evaluation, and optimization

Job Summary

  • Build and operate scalable machine learning and data systems, simulation workflow and insight tools to improve and speed up evaluation and onboard developer journeys.
  • Provide deep technical leadership on large-scale ML model architectures, especially for autonomous vehicle models, and build scalable systems for training and fine-tuning large-scale models.
  • Design and scale large distributed systems covering the ML lifecycle, supporting planet-scale dataset generation, model training, and evaluation.

Matching Summary

Match Score: 85

Waymo is seeking a Staff Machine Learning Software Engineer to lead the development of scalable machine learning systems for their autonomous vehicle technology. The ideal candidate will have extensive experience in machine learning infrastructure, with a strong focus on model training, evaluation, and optimization.

Salary

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

Skills & Requirements

Must-have

  • large-scale ML model architectures
  • scalable systems for training
  • data engineering and model development
  • distributed systems for ML lifecycle
  • autoregressive transformers
  • PyTorch, Jax, Tensorflow, Ray

Nice-to-have

  • autonomous vehicles domain experience
  • robotics or complex simulation environments
  • RL techniques for fine-tuning
  • large-scale simulation platforms
  • metrics for evaluating complex AI systems
  • technical leadership and influencing stakeholders

Key Requirements

  • M.S. or Ph.D. degree or equivalent practical experience
  • 5+ years professional software engineering
  • 3+ years ML infrastructure experience
  • Strong expertise in distributed training
  • Experience with ML accelerator profiling tools
  • Strong leadership skills

Work Rights

Not specified

Tailored Resume

Cover Letter