Staff / Senior Staff Software Engineer - Learned Trajectory Machine Learning Engineer
Develops deep learning models using imitation and reinforcement learning to generate safe, efficient driving trajectories for autonomous vehicles. Collaborates with Perception, Planning, and Simulation teams; requires ML expertise, Python fluency, and transformer experience.
About the job
Responsibilities
- Develop new deep learning models using imitation learning and reinforcement learning to generate driving plans for autonomous vehicles.
- Work on techniques to estimate the quality of driving plans along dimensions of safety, progress, comfort, etc.
- Leverage 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 systems.
- Collaborate with engineers on Perception, Planning, and Simulation to solve the overall Autonomous Driving problem in complex urban environments.
Qualifications
- BS, MS, or PhD degree in computer science or related field.
- Experience with training and deploying transformer-based model architectures and reinforcement learning.
- Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines.
- Fluency in Python with a basic understanding of C++.
- Extensive experience with programming, algorithm design, and strong mathematics skills.
Bonus Qualifications
- Conference or Journal publications in Machine Learning or Robotics related venues.
- Prior experience with Prediction and/or autonomous vehicles or robotics in general.
Skills
Python, C++, Transformers, Reinforcement Learning, Imitation Learning, Machine Learning Pipelines, Deep Learning, Mathematics, Autonomous Driving, Robotics
Similar jobs
ML Engineering jobs
Square
San Francisco, CA
Build and operate production machine learning systems for ranking, retrieval, recommendations, personalization, and customer intelligence. The role requires 12+ years of production software and ML experience, strong expertise in intelligent systems, and sound judgment around trustworthy customer-impacting signals.
United States
Leads the technical direction and development of large-scale, GenAI-powered recommendation and feed-ranking systems. Requires 10+ years of industry experience in relevance-driven products, deep expertise in machine learning and recommendations, and strong organizational influence and mentoring skills.
United States
Leads technical strategy for Reddit’s Ads ML Platform, improving feature development, training-data generation, experimentation, and the path to production ML serving. The role requires 8+ years in infrastructure or distributed systems, production ML platform experience, and strong cross-team technical leadership.
Garner Health
New York, NY
Leads the reliability, architecture, deployment automation, and monitoring of production machine learning systems. Requires 7+ years of software engineering experience, deep MLOps platform expertise, and strong Kubernetes, cloud, infrastructure-as-code, and observability fundamentals.
Garner Health
New York, NY
Leads end-to-end development of production algorithmic systems for healthcare, spanning machine learning, optimization, and LLM applications. The player-coach role requires 6+ years of industry experience, strong problem-solving and metrics judgment, and technical leadership of a small team.