Senior/Staff Machine Learning Engineer (Reinforcement Learning, Motion Planning)

Join our team as a Senior/Staff Machine Learning Engineer specializing in Reinforcement Learning and Motion Planning. In this role, you will design, develop, and deploy learned behavior planning models, own the end-to-end Reinforcement Learning pipeline, and contribute to our Reinforcement Learning Infrastructure.

Responsibilities

• You will collaborate with simulation and planning teams to deploy learned behavior planning models to driving pipelines.
• Enjoy benefits such as unlimited PTO, flexible working arrangements, health benefits, equity options, professional development opportunities, and daily catered lunches.
• Concevoir, développer et déployer des modèles de planification de comportement appris en utilisant des approches telles que l'apprentissage par renforcement.
• Posséder l'ensemble du pipeline d'apprentissage par renforcement, de la construction de l'environnement à l'entraînement et à la modélisation.
• Collaborer avec les équipes de simulation et de planification pour déployer des modèles de planification de comportement appris.

Requirements

• 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
• Strong oral and written communication skills
• Experience with reinforcement learning
• Experience with self-play reinforcement learning

Preferred

• Experience with high-throughput batched driving simulators such as GPUDrive and PufferDrive
• Experience with sim-to-real in the context of robotics / autonomous driving
• Prior experience in learned behavior planning / prediction for autonomous vehicles

Pay

USD 130,000 - 220,000