Waymo

Staff Machine Learning Engineer – VLM/LLM Evaluation

Waymo
Apply
7 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

On the journey to be the world's most trusted driver. With the Waymo Driver, we can improve mobility while saving thousands of lives. Download the Waymo One app and ride today. Waymo reaches out to candidates from official channels only (e.g. directly from @waymo.com email addresses, or through our recruiters or sourcers who are noted as such on LinkedIn). We do not contact candidates about career opportunities through instant messaging apps like Telegram, email addresses from domains other than waymo.com (such as Gmail addresses), direct messages on Twitter, Facebook, and Instagram, or text messages. Visit waymo.com to check out our official job listings. Need support? Send us a message – we're here to help.

Contact me