P
PlusAI Verified
Automotive, Autonomous Driving, Artificial Intelligence, Logistics Technology
Senior/Staff Machine Learning Engineer (Reinforcement Learning), Motion Planning
Santa Clara, CAHybridFull TimeStaffPosted 2 months agoVisa sponsorship available
Required skills
- Reinforcement Learning
- Machine Learning
- C++
- Python
- PyTorch
- Robotics
- Computer Engineering
- Software Engineering
Nice to have
- GPUDrive
- PufferDrive
- self-play reinforcement learning
- sim-to-real
- autonomous driving
- learned behavior planning
Role summary
PlusAI is seeking a Senior/Staff Machine Learning Engineer specializing in Reinforcement Learning for its autonomous truck division. This role involves designing, developing, and deploying learned behavior planning models using RL, behavior cloning, and imitation learning. You will own the end-to-end RL pipeline, contribute to RL infrastructure for large-scale distributed training, and analyze model performance. The position requires experience in C++, Python, PyTorch, and reinforcement learning, with an MS or PhD in a related field. Preferred qualifications include experience with high-throughput driving simulators, self-play RL, and sim-to-real applications in robotics or autonomous driving.
PlusAI is a Physical AI company pioneering AI-based virtual driver software for factory-built autonomous trucks. Headquartered in Silicon Valley with operations in the United States and Europe, Plus was named by Fast Company as one of the World’s Most Innovative Companies. Partners including TRATON GROUP’s Scania, MAN, and International brands, Hyundai Motor Company, Iveco Group, Bosch, and DSV are working with Plus to accelerate the deployment of next-generation autonomous trucks. If you’re ready to make a huge impact and drive the future of autonomy, Plus is looking for talented individuals to join its fast-growing teams. 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 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. 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 Your opportunities joining PlusAIWork, learn and grow in a highly future-oriented, innovative and dynamic field.Wide range of opportunities for personal and professional development.Catered free lunch, unlimited snacks and beverages.Highly competitive salary and benefits package, including 401(k) plan.
Responsibilities
Required skills
Preferred skills
Similar roles
- PData Engineer (SE / Sr SE)PlusAI · Santa Clara, CA · Hybrid
- PFleet ManagerPlusAI · Fremont, CA · Onsite
- PSr. Machine Learning EngineerPinterest Job Advertisements · Remote · Remote
- Machine Learning Engineer - Income InsightsPlaid · New York, NY · Hybrid
Senior Machine Learning EngineerClio · Burnaby, BC · Hybrid
Senior Machine Learning Engineer, Perception - Autonomous DrivingNVIDIA · Remote · Onsite
