Research Scientist, RL for Autonomous Planning & World Modeling

3 months, 3 weeks ago
Full-time
Senior
Software Development

Waymo

Waymo is an autonomous driving technology company building the Waymo Driver and operating Waymo One, its fully autonomous ride-hailing service.

Autonomous vehicles, robotics, AI, ride-hailing / mobility tech
Founded 2009
$21600M raised

Description

  • Participate in Waymo’s Foundation World Model post-training and evaluation.
  • Research and develop reinforcement learning and distillation techniques for autonomous vehicle trajectory planning.
  • Integrate emerging research from the broader AI community into Waymo’s internal reinforcement learning infrastructure.
  • Run rigorous ablation studies to identify and scale the most promising methods.
  • Partner with engineering and research teams across Waymo to share recipes, techniques, and post-training best practices.

Requirements

  • PhD or Master’s in Computer Science, Machine Learning, Robotics, or a similar technical field.
  • 3+ years of industry or post-doc research experience in Reinforcement Learning or Foundation Models.
  • Demonstrated original contributions through high-impact publications, technical blog posts, or significant open-source contributions.
  • Proficiency implementing model training flows in a scalable, distributed, and performant manner, including data parallel, FSDP, and other sharding approaches.
  • Willingness to work with the complexity of globally distributed inference infrastructure.
  • Preferred: PhD in Computer Science, Machine Learning, or Robotics with a research focus on Reinforcement Learning, Foundation Models, or Multi-Modal learning.
  • Preferred: Extensive experience designing and deploying reinforcement learning infrastructure, especially for on-policy learning or alignment with human preferences.
  • Preferred: First-author publications at top-tier venues such as NeurIPS, ICLR, or ICRA, or significant open-source ML projects.
  • Preferred: Experience with large-scale many-machine training infrastructure and inference techniques for large models such as model sharding or tensor-parallel.
  • Remote work is possible, but Waymo may not be able to employ remotely in all locations.

Benefits

  • Base salary range of $204,000 to $259,000 USD.
  • Eligibility for Waymo’s discretionary annual bonus program.
  • Eligibility for Waymo’s equity incentive plan.
  • Access to Waymo’s generous company benefits program, subject to eligibility requirements.
  • Remote work may be available depending on location.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Senior Security Research Engineer

PlayStation 100K+ Household Durables

Sony Interactive Entertainment is hiring a Senior Security Research Engineer to lead Global Vulnerability Management efforts that advance its Threat Exposure Management program and strengthen security risk reduction across the PlayStation ecosystem.

Bash CI/CD Domo Git JavaScript PowerShell Python Snowflake SQL
2 days, 16 hours ago

Research Engineer

Weekday 11-50 Construction & Engineering

Weekday is hiring a Research Engineer for one of its clients to help develop next-generation AI/ML voice agents for production conversational systems in India.

AWS Azure Deep Learning GCP Generative AI Hugging Face LLM Machine Learning NLP Python PyTorch TensorFlow Transformers
2 days, 16 hours ago

Senior Research Scientist, Evidence Synthesis

Precision AQ 1001-5000 Business Consulting and Services

Precision Medicine Group is hiring a Senior Research Scientist to lead evidence synthesis projects supporting health technology assessment submissions in a fully remote role based in Canada or the US.

R
5 days, 15 hours ago

Senior Biological Data Scientist

bridgebiowp 251-1K Pharmaceuticals

GondolaBio, a BridgeBio affiliate, is hiring a U.S.-based remote computational biologist to run integrative multiomics analyses that accelerate therapeutic discovery in rare and genetically defined diseases.

LLM Python
6 days, 15 hours ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers