Skip to content

Research Scientist - Reinforcement Learning, Self-Driving

126k – 423kSunnyvale, CAOnsite
Summary

Conducts cutting-edge research in reinforcement learning, self-play RL, VLA post-training, and closed-loop RL for autonomous driving and robotics. Requires strong research record with publications, MSc/PhD in ML/CV, and expertise in Python, PyTorch, computer vision, and robotics.

About the role

Responsibilities

  • Conduct research on reinforcement learning (RL) related topics including large-scale self-play RL, VLA post-training, large-scale closed-loop RL based on neural simulation with applications to autonomous driving
  • Dive into fundamental topics on RL with broader applications and potential imitative behavior learning incorporation, and relevant topics such as reward learning
  • Work closely with other Research Scientists and interns on research publications for submission to top-tier conferences
  • Collaborate with Research Engineers and engineering teams to test and deploy algorithms to our autonomy and robotics products

Requirements

  • Strong research record in the fields of RL and VLA post-training for autonomous systems and robotics, with publications in top-tier conferences or journals in the fields of computer vision, machine learning, and robotics
  • MSc or PhD in machine learning and computer vision with autonomy and robotics applications or closely-related fields
  • Passion for next-generation, scalable autonomy and robotics for real-world systems
  • Strong research skills and the ability to work both independently and collaboratively on projects
  • Technical experience in: Python, PyTorch, computer vision, robotics systems, and distributed machine learning model training

Nice to Have

  • Hands-on experience in at least one of the following fields:
    • Self-play RL and imitation learning, behavior learning
    • VLA post-training for autonomy or robotics
    • Large-scale closed-loop RL in driving simulation
    • Large-scale RL training infrastructure (Ray preferred)

Compensation

  • Base salary range: $126,000 - $423,000 USD annually
  • Equity, comprehensive health/dental/vision/life/disability insurance, 401k with employer match, learning/wellness stipends, paid time off
Skills
Reinforcement LearningPyTorchPythonComputer VisionRoboticsDistributed Machine LearningSelf-play RLImitation LearningVLA post-trainingRay
Similar roles at this salary range
All ML Engineering jobs →
Chime

AI/ML Engineer

Build and productionize ML models for risk detection and decisioning systems. Requires 1-2 years applied ML experience and familiarity with AWS, model evaluation, and experimentation.

125k – 173kSan Francisco, CAML EngineeringHybridAWSPython
PrizePicks

Machine Learning Platform Engineer

Build and operate the ML platform to productionize models, enable real-time inference, and manage the full ML lifecycle with MLOps best practices. Requires 3+ years platform engineering and 1+ years owning ML systems end-to-end.

135k – 160kUnited StatesML EngineeringRemoteGoKafka
Ai2

Senior Research Engineer

Research Engineer training and scaling flagship open multimodal and agentic models (Olmo, Molmo). Owns end-to-end ML infrastructure, model development, and open-source releases.

147k – 220kSeattle, WAML EngineeringOn-siteJAXMoE
Clickhouse

AI Product Engineer

Build agentic capabilities on a petabyte-scale observability platform. Own the full agent stack including context engineering, tool design, evals, and production reliability for incident investigation.

130k – 230kUnited StatesML EngineeringRemoteMCPSQL
Kodex

ML Engineer

Build and deploy production ML models and pipelines to detect suspicious activity, improve verification accuracy, and support threat intelligence workflows.

150k – 180kUnited StatesML EngineeringRemoteAWSClustering