GrepJob
Waymo

Machine Learning Engineer (Infra), Driver Understanding and Evaluation

Waymo
Apply
about 8 hours ago
Mountain View, CA, USA
Mid Level / Senior
H1B Sponsor

Base Salary

$170k - $216k/yr

Responsibilities

  • Build scalable systems for training and fine-tuning large-scale models to evaluate driving behaviors.
  • Provide guidance on architectural decisions and own large, complex systems.
  • Contribute to the production and optimization of machine learning models for Waymo's fleet.
  • Design and scale distributed systems covering the ML lifecycle.
  • Collaborate cross-functionally to derive performance and system-level requirements.

Requirements

  • M.S. or Ph.D. in Computer Science, Machine Learning, AI, or related field.
  • 3+ years of experience in machine learning infrastructure.
  • History of contributions to machine learning tooling and frameworks.
  • Strong expertise in distributed training techniques.
  • Understanding of user-facing APIs and internal workings of ML frameworks.

Benefits

  • Eligible for Waymo's discretionary annual bonus program.
  • Participation in equity incentive plan.
  • Generous Company benefits program.

Tech Stack

PyTorchTensorFlow

Categories

AI & MLData EngineeringData Science