Waymo

Staff Machine Learning Engineer – VLM/LLM Evaluation

Waymo
Apply
8 months ago
Mountain View, CA, USA +2 moreStaff+
H1B sponsor

Base Salary

$238k - $302k/yr

Responsibilities

  • Lead development of end-to-end evaluation systems and benchmarks for Waymo foundation models across pretraining, supervised fine-tuning, and reinforcement learning.
  • Evaluate the quality, safety, and realism of embodied AI agents.
  • Implement and extend large-scale data and evaluation pipelines.
  • Collaborate within and across organizations to bring innovative technology into production.
  • Help build foundation models used onboard autonomous vehicles and offboard in simulation.

Requirements

  • Master’s degree or PhD in Computer Science, a similar technical field, or equivalent practical experience.
  • At least 5 years of experience in ML engineering and applied deep learning.
  • Strong portfolio of shipped products or a publication record.
  • Experience with large-scale distributed systems.
  • Proficient programming skills in Python and C/C++.
  • Strong analytical and debugging skills.
  • Preferred experience training, evaluating, and deploying ML models at scale.
  • Preferred deep learning experience with generative models such as LLMs/VLMs and/or reinforcement learning.
  • Preferred in-depth knowledge of an ML framework such as PyTorch, JAX, or TensorFlow.

Benefits

  • Hybrid work schedule.
  • Full-time position.
  • Eligibility for discretionary annual bonus, equity incentive plan, and company benefits program, subject to eligibility requirements.

Tech Stack

CC++PythonPyTorchTensorFlow

Categories

Waymo

About Waymo

1,001-5,000 employees

Waymo develops the Waymo Driver, a full-stack autonomous driving system, and operates the Waymo One robotaxi service that offers paid, driverless rides in select U.S. cities. The company monetizes through ride-hailing fares and partnerships integrating its technology on multiple vehicle platforms. Founded in 2009 as Google’s self-driving project, Waymo is headquartered in Mountain View and is an Alphabet subsidiary, with over 10 million rider-only trips and 100+ million autonomous miles on public roads.

Contact me