Senior ML Engineer — RL Pipelines for Autonomous Driving

Plus 2; Santa Clara (CA); On-site; USD 120,000 - 180,000; Full time

Benefits offered by this job

Catered lunch

Unlimited snacks and beverages

401(k) plan

Job summary

PlusAI is a Silicon Valley–based Physical AI company pioneering AI-based virtual driver software for factory-built autonomous trucks. We design, develop, and deploy RL-driven planning models and scalable training systems to push the boundaries of autonomous driving.

Join a fast-growing team focused on high-performance, production-grade ML in a robotics context, with opportunities to collaborate across simulation and planning groups and deploy models into real driving pipelines.

Qualifications

• 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.

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.

Skills

C/C++

Python

Reinforcement Learning

PyTorch

Communication skills

Education

MS/PhD in CS/Robotics/ML

Tools

GPUDrive

PufferDrive

Job description

PlusAI is a Silicon Valley–based Physical AI company pioneering AI-based virtual driver software for factory-built autonomous trucks. We design, develop, and deploy RL-driven planning models and scalable training systems to push the boundaries of autonomous driving.

Join a fast-growing team focused on high-performance, production-grade ML in a robotics context, with opportunities to collaborate across simulation and planning groups and deploy models into real driving pipelines.