Waymo · 1 month ago
Staff Machine Learning Engineer, Infrastructure
Mountain View, CA, USA
Full-time
Onsite
Senior Level, Lead/Staff
$238K/yr - $302K/yr
7+ years expWaymo is an autonomous driving technology company on a mission to be the world's most trusted driver. The Staff Machine Learning Engineer, Infrastructure will collaborate with various teams to develop and maintain CI/CD pipelines, build data management tooling, and ensure system reliability in a rapidly evolving environment.
Artificial Intelligence (AI)AutomotiveAutonomous VehiclesSensorTransportation
Responsibilities
Collaborate with machine learning engineers, data scientists, and infrastructure engineers to surface signals on model, component, and system performance
Develop, scale, and maintain CI/CD pipelines for automating metric computation, aggregation, and monitoring
Build data management tooling to improve reproducibility and scale analyses automatically as Waymo drives more miles
Collaborate with ML and product teams to define evaluation criteria and translate those into metrics and targets
Stay current with emerging technologies and trends in ML evaluation and metrics design
Be aware of best practices used in the Alphabet stack of ML technologies (e.g. TF, JAX, Flax, Beam etc)
Qualification
Required
B.S. in Computer Science, Math, or equivalent real-world experience
7+ years building and maintaining high-scale distributed or ML inference systems
Coding and testing skills, specifically Python/C++
Familiarity with large-scale fleet management, testing, and deployment, e.g.: GCP/AWS/Kubernetes, Jenkins, GCS/S3, etc
Familiarity with large-scale ML tools, e.g.: tf.serving/Torchserve, Kubeflow/Sagemaker Pipelines/Vertex AI Pipelines, etc
Understanding of machine learning fundamentals and experience with popular ML frameworks such as JAX, PyTorch, or TensorFlow
Preferred
Experience developing and maintaining evaluation pipelines for ML models
Experience deploying and supporting machine learning models for computer vision, natural language processing, robotics/motion planning, or recommendation systems
Experience supporting a large team of MLEs developing high-capacity, production-grade models and components
Strong understanding of metrics computation and regression detection at scale
Benefits
Waymo’s discretionary annual bonus program
Equity incentive plan
Generous Company benefits program
Company
Waymo
Waymo is a mobility technology company that improves transportation by developing self-driving solutions for travelers and daily commuters. It is a sub-organization of Alphabet.
1001-5000 employees
H1B Sponsorship
Waymo has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (231)
2024 (175)
2023 (268)
2022 (306)
2021 (298)
2020 (317)
Funding
Current Stage
Late StageTotal Funding
$11.1BKey Investors
Alphabet
2024-07-23Series C· $5.6B
2021-06-16Series B· $2.5B
2020-05-12Series A· $750M
Leadership Team
Tekedra Mawakana
Co-Chief Executive Officer
Elisa de Martel
Chief Financial Officer
Recent News
2026-01-12
PYMNTS.com
2026-01-12
Company data provided by crunchbase