Senior/Staff Machine Learning Engineer (Reinforcement Learning), Motion Planning
Responsibilities
- Design, develop, and deploy learned behavior planning models using approaches such as reinforcement learning, behavior cloning, and imitation learning
- Own end-to-end Reinforcement Learning pipeline from environment construction, training, and modeling.
- Contribute to Reinforcement Learning Infrastructure to facilitate large-scale distributed training
- Track model training, analyze model performance, identify failure modes to improve generalization across driving scenarios
- Collaborate with simulation and planning teams to deploy learned behavior planning models to driving pipelines
Required skills
- MS or PhD in Computer/Software Engineering, Robotics, Machine Learning or related field.
- Experience developing production code in C/C++ and Python
- Experience with machine learning frameworks such as PyTorch
- Experience with reinforcement learning
- Strong oral and written communication skills.
Preferred skills
- Experience with high-throughput batched driving simulators such as GPUDrive and PufferDrive
- Experience with self-play reinforcement learning
- Experience with sim-to-real in the context of robotics / autonomous driving
- Prior experience in learned behavior planning / prediction for autonomous vehicles
Skills
As published by lever · 3 questions · 3 written answers
Resume/CV, Full name, Email, Phone, Current location, Current company, LinkedIn URL, GitHub URL
- Are you legally authorized to work in the US
- Will you require visa sponsorship now or in the future? If so, please specify.
- How did you hear about this opportunity? optional