Skip to content

Research Engineer - Reinforcement Learning, Self-Driving

126k – 423kSunnyvale, CAOnsiteEntry level
Summary

Conducts research on reinforcement learning for self-driving cars and robotics, develops large-scale RL training infrastructure, and deploys algorithms to production systems. Requires hands-on RL experience, PyTorch/Python proficiency, and strong research skills.

About the role

Responsibilities

  • Conduct research on reinforcement learning (RL) and its training infrastructure, including large-scale self-play RL, VLA post-training, large-scale closed-loop RL based on neural simulation with applications to autonomous driving.
  • Work closely with Research Scientists and interns on high-quality research publications to submit to top-tier conferences.
  • Collaborate with engineering teams on ADAS, data, and simulation to deploy end-to-end algorithms for mass production vehicles, and neural simulation/generation for tools supporting autonomy development.

Requirements

  • Hands-on experience in at least one of: 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).
  • Passion for next-generation, scalable autonomy and robotics for real-world systems.
  • Strong engineering and research skills and ability to work independently and collaboratively.
  • Technical experience in: Python, PyTorch, computer vision, robotics systems, and distributed machine learning model training.

Nice to Have

  • Industry experience on relevant topics (self-driving application preferred).
  • MSc or PhD in machine learning and computer vision with autonomy and robotics applications or closely related field.
  • Passion for building and shipping customer-focused software frameworks or tools.

Compensation

Base salary range: $126,000 - $423,000 USD annually, plus equity and benefits.

Skills
Reinforcement LearningPyTorchPythonComputer VisionRayImitation LearningDistributed Machine LearningRoboticsSelf-Play RLVLA
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