Waymo Logo

Waymo

Senior Machine Learning Engineer, Growth

Posted 4 Days Ago
Be an Early Applicant
Hybrid
2 Locations
175K-215K Annually
Senior level
Hybrid
2 Locations
175K-215K Annually
Senior level
Build and own production machine learning infrastructure for growth initiatives, including feature pipelines, point-in-time-correct training data, batch inference, model registries, retraining automation, validation safeguards, and production monitoring. Partner closely with data science and data engineering to productionize models, prevent data leakage and train-serve skew, and ensure reliable model-driven promotions.
The summary above was generated by AI

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced Driver™—to improve access to mobility while saving thousands of lives now lost to traffic crashes. The Waymo Driver powers Waymo’s fully autonomous ride-hail service and can also be applied to a range of vehicle platforms and product use cases. The Waymo Driver has provided over ten million rider-only trips, enabled by its experience autonomously driving over 100 million miles on public roads and tens of billions in simulation across 15+ U.S. states.

The Growth Machine Learning Engineering team partners closely with Data Science to take models from experimentation into reliable production. We own the underlying ML data infrastructure: standardizing shared feature definitions, eliminating data leakage in training pipelines, automating scheduled retraining, and establishing safeguards so faulty batch scoring runs never affect live promotions. By managing data integrity, deployment pipelines, and production monitoring, we provide the dedicated engineering support needed to keep model-driven growth initiatives dependable and scalable.

You will:

  • Own the feature layer between the Growth DB and the models. Work with data engineering on schema design for ML features, and with data science to turn their feature lists into versioned feature definitions that training and scoring both read from.
  • Build point-in-time correct training data, so a retrain sees each day as it was and does not leak future data. Data science has already named this as a gap in today's pipelines.
  • Take models data science has trained and put them into production: validate against the offline results, wire them into batch scoring, and add input and output checks that stop a bad run before promos go out.
  • Lead training automation: scheduled retrains, a model registry, and validation before a new model version is promoted.
  • Monitor models in production: feature drift, score distribution shifts, and model versus random arm results.
  • Be data science's day-to-day engineering counterpart, so feature and modeling questions do not queue behind the tech lead.

You have:

  • Experience at roughly building production ML systems: feature pipelines, training data generation, batch inference.
  • Strong SQL and Python, plus large-scale data processing (Flume, Beam, Spark or similar).
  • Designed feature schemas or worked on a feature store, and understands point-in-time correctness and train/serve skew from having fixed them.
  • Productionized and owned the lifecycle of an existing offline model
  • Worked with data scientists and can turn a loose feature request into a spec.

We prefer:

  • Background in uplift modeling, causal inference or experimentation (A/B tests, holdouts), enough to understand what the models need.
  • Experience in ads, marketing, recommendations, pricing or promotions.
  • Familiarity with feature stores, workflow orchestration (Airflow or similar) and distributed SQL databases.

#LI-Hybrid

The expected base salary range for this full-time position across US locations is listed below. Actual starting pay will be based on job-related factors, including exact work location, experience, relevant training and education, and skill level. Your recruiter can share more about the specific salary range for the role location or, if the role can be performed remote, the specific salary range for your preferred location, during the hiring process. 

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. 

Salary Range
$175,000—$215,000 USD
HQ

Waymo Mountain View, California, USA Office

1600 Amphitheatre Pkwy, Mountain View, CA, United States, 94043

Waymo San Francisco, California, USA Office

San Francisco, United States

Similar Jobs

One Month Ago
In-Office or Remote
94104, San Francisco, CA, USA
171K-269K Annually
Senior level
171K-269K Annually
Senior level
Cloud • Information Technology • Productivity • Security • Software • App development • Automation
Build and operate scalable machine learning systems for personalized recommendations, growth journeys, cross-product expansion, and sales experiences. Design models, datasets, features, evaluation frameworks, and decisioning services; run offline policy evaluations and online experiments; and monitor attribution, latency, quality, and fairness. Partner with product, engineering, data science, analytics, marketing, and sales teams to turn ambiguous growth opportunities into measurable production capabilities and strategic ML initiatives.
Top Skills: SparkAWSDatabricksJavaPythonSQLTypescript
One Month Ago
In-Office or Remote
94104, San Francisco, CA, USA
171K-269K Annually
Senior level
171K-269K Annually
Senior level
Cloud • Information Technology • Productivity • Security • Software • App development • Automation
Build and operate scalable machine learning systems for Atlassian’s Growth organization, including personalized recommendations, ranking, decisioning, experimentation, and customer journey optimization. Partner with product, engineering, data science, analytics, marketing, and sales teams to translate growth opportunities into measurable production capabilities. Develop datasets, features, models, evaluation frameworks, and monitoring systems while optimizing for customer value, fairness, quality, and long-term business outcomes.
Top Skills: SparkAWSDatabricksJavaPythonSQLTypescript
One Month Ago
Remote or Hybrid
Mountain View, CA, USA
166K-273K Annually
Senior level
166K-273K Annually
Senior level
AdTech • Artificial Intelligence • Gaming • Machine Learning • Software • Virtual Reality • Metaverse
Design and lead large-scale machine learning and backend systems for advertiser growth, including budget pacing algorithms, GenAI creative generation infrastructure, marketplace experimentation, high-throughput billing pipelines, and real-time stream processing. Analyze marketplace and financial data, optimize advertiser performance, ensure billing accuracy and fault tolerance, lead multi-quarter technical roadmaps, and mentor senior engineers.
Top Skills: A/B TestingDiffusion ModelsFlinkGenerative AiGoJavaKafkaLarge Language Models (Llms)Pid ControllersScalaSparkStream Processing

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account