Machine Learning Engineer
Develop deep learning models using imitation and reinforcement learning for autonomous driving agents and planning. Requires advanced degree plus experience with RL, transformers, and production ML pipelines.
About the job
Responsibilities
- Develop new deep learning models that use imitation learning and reinforcement learning to generate driving plans for human-like agents.
- Work on novel techniques to estimate the quality of those driving plans along the dimensions of safety, progress, comfort and realism.
- Contribute to our large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field.
- Develop metrics and tools to analyze errors and understand improvements of our systems.
- Collaborate with engineers on Perception, Planning, Simulation, and Validation to solve the overall Autonomous Driving problem.
Requirements
- PhD degree in computer science or related field or master's degree and 5+ years of professional experience in a relevant field.
- Experience in Planning and / or Prediction using Reinforcement Learning techniques.
- Experience with training and deploying transformer-based model architectures.
- Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines.
- Fluency in Python with a basic understanding of C++.
Nice-to-Haves
- Top tier publications (NeurIPS, ICML, CVPR).
Skills
Reinforcement Learning, Imitation Learning, Transformers, Python, C++, Machine Learning Pipelines, Planning, Prediction, Deep Learning
Similar jobs
ML Engineering jobs
Earnin
Mountain View, CA
Build production-grade AI agents, evaluation infrastructure, and developer tooling that make AI-assisted engineering faster, safer, and reusable across teams. The role requires software engineering experience, platform or internal developer-product experience, and hands-on expertise with LLM integration and orchestration.
Nuro
Mountain View, CABuild trustworthy infrastructure for production LLM agents, closed-loop evaluation, and autonomous research workflows. The role requires strong Python and distributed-systems experience, hands-on LLM post-training and inference knowledge, and experience operating agent systems at scale.
ClickUp
United States
Build and operate large-scale ranking and retrieval systems that power search relevance, including hybrid lexical/vector search, embeddings, query understanding, and permission-aware retrieval. Requires a bachelor's degree and 5+ years of ML engineering experience in ranking or information retrieval.
Atomicmachines
Emeryville, CA
Build and operate production ML infrastructure spanning training, deployment, serving, monitoring, data pipelines, and feedback-driven retraining. The role requires strong MLOps and DevOps experience, Python and SQL proficiency, and ownership of reliable cloud-based systems.
Tessera Labs
San Jose, CABuild and scale post-training, reinforcement-learning, evaluation, and inference systems for long-horizon agents operating over complex enterprise software. The role requires strong Python and PyTorch or JAX skills, distributed GPU experience, empirical rigor, and the ability to take research results into production.