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

PlusAI, Inc.; Santa Clara (CA); Hybrid; USD 180,000 - 250,000; Full time

Benefits offered by this job

Catered free lunch

Unlimited snacks and beverages

Competitive salary and benefits, incl.

401(k) plan

Job summary

PlusAI, Inc. is a Silicon Valley–based Physical AI company pioneering AI-based virtual driver software for autonomous trucks.

We operate in the United States and Europe, partnering with leading brands to accelerate autonomous trucking adoption. The Senior/Staff Machine Learning Engineer (Reinforcement Learning, Motion Planning) will design, develop, and deploy learned behavior planning models, own end-to-end RL pipelines, and contribute to scalable RL infrastructure.

Qualifications

• MS or PhD in Computer/Software Engineering, Robotics, ML 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.

Responsibilities

• Design, develop, and deploy learned behavior planning models with RL, behavior cloning, and imitation learning.
• Own end-to-end RL pipeline: environment construction, training, and modeling.
• Contribute to RL infrastructure for large-scale distributed training.
• Track training, analyze performance, and improve generalization across driving scenarios.
• Collaborate with simulation/planning teams to deploy models into driving pipelines.

Skills

C/C++

Python

PyTorch

Reinforcement Learning

Communication skills

Education

MS or PhD in Computer/Software Engineering, Robotics, Machine Learning or related field

Tools

C/C++

Python

PyTorch

Job description

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

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

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

Work, 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.

Whether you need an AI driver for your next vehicle program or a platform
to build autonomy you control, PlusAI meets you where you are.