Senior ML Engineer — RL Motion Planning for Autonomous Trucks

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

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.