Senior ML Engineer: Learned Driving & RL Pipelines

PlusAI; Santa Clara (CA); On-site; USD 130,000 - 220,000; Full time

Benefits offered by this job

Catered lunch

Unlimited snacks

401(k) plan

Job summary

PlusAI in Silicon Valley seeks a researcher to design, implement, and deploy learned behavior planning models for autonomous trucks. You will own end-to-end RL pipelines from environment construction to modeling and deployment.

Candidates have strong ML, C/C++, Python and PyTorch experience, plus familiarity with RL. The role offers a competitive salary and a comprehensive benefits package including 401(k) and lunch.

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.

Responsibilities

• Design, develop, and deploy learned behavior planning models using 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

PyTorch

Reinforcement learning

Communication skills

Education

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

Tools

GPUDrive

PufferDrive

Job description

PlusAI in Silicon Valley seeks a researcher to design, implement, and deploy learned behavior planning models for autonomous trucks. You will own end-to-end RL pipelines from environment construction to modeling and deployment.

Candidates have strong ML, C/C++, Python and PyTorch experience, plus familiarity with RL. The role offers a competitive salary and a comprehensive benefits package including 401(k) and lunch.