Senior Machine Learning Engineer - ML Agents and Planning
Develops ML models using imitation and reinforcement learning for agent behavior prediction and autonomous vehicle planning. Requires PhD +1 year or MSc +5 years experience, expertise in RL, transformers, and production ML pipelines.
About the job
Responsibilities
- Develop deep learning models using imitation learning and reinforcement learning for human-like agent driving plans.
- Create novel techniques to estimate driving plan quality (safety, progress, comfort, realism).
- Contribute to large-scale ML infrastructure to discover solutions and advance the field.
- Develop metrics and tools to analyze errors and improvements.
- Collaborate with Perception, Planning, Simulation, and Validation engineers on autonomous driving.
Qualifications
- PhD in computer science or related field +1 year professional experience (top-tier publications may substitute), or MSc +5 years professional experience.
- Experience in Planning/Prediction using Reinforcement Learning.
- Experience training and deploying transformer-based models.
- Experience with production ML pipelines: dataset creation, training frameworks, metrics.
- Fluency in Python; basic C++ understanding.
Bonus Qualifications
- Top-tier publications (NeurIPS, ICML, CVPR).
Skills
Reinforcement Learning, Imitation Learning, Transformers, Python, C++, Machine Learning Pipelines, Deep Learning, Planning, Prediction, Autonomous Driving
Similar jobs
ML Engineering jobs
Baselayer
San Francisco, CA
Senior AI Engineer responsible for production LLM agents that enrich business identity data through web discovery, verification, classification, and risk scoring. The role requires strong asynchronous Python, agent and evaluation expertise, browser automation, and experience operating AI systems in production.
Otter
Mountain View, CA
Leads the development and production deployment of large-scale ASR and TTS systems for conversational intelligence products. The role requires 5+ years of industry experience, deep speech-model expertise, and strong software engineering and ML operations capabilities.
Ambience Healthcare
San Francisco, CA
Build and improve production AI systems for clinical products, owning evaluations, model behavior, agentic workflows, data flywheels, deployment, and observability. The role requires 5+ years of production ML or applied AI experience, strong Python and modern ML framework skills, and hands-on debugging expertise.
Dialpad
United States
Leads development of speech models, decoders, and low-latency inference systems for next-generation voice agents. Requires 5+ years in speech ML or related audio AI, strong Python and PyTorch experience, and the ability to guide technical direction and mentor engineers.
Ontario, Canada
Design, build, and deploy production ML systems for recommendations, search, ranking, and advertising at internet scale. Own the full ML lifecycle from modeling to monitoring with strong cross-functional collaboration.