2 months ago
Santa Clara, CA, USASenior / Staff+
H1B sponsor
Base Salary
$130k - $220k/yr
Responsibilities
- Design, develop, and deploy learned behavior-planning models using reinforcement learning, behavior cloning, and imitation learning.
- Own the end-to-end reinforcement learning pipeline, including environment construction, training, and modeling.
- Contribute to reinforcement learning infrastructure for large-scale distributed training.
- Track training, analyze model performance, identify failure modes, and improve generalization across driving scenarios.
- Collaborate with simulation and planning teams to deploy learned behavior-planning models to driving pipelines.
Requirements
- Required: MS or PhD in Computer Engineering, Software Engineering, Robotics, Machine Learning, or a 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: experience with high-throughput batched driving simulators such as GPUDrive and PufferDrive.
- Preferred: experience with self-play reinforcement learning and sim-to-real in robotics or autonomous driving.
- Preferred: prior experience in learned behavior planning or prediction for autonomous vehicles.
Benefits
- Highly competitive salary and benefits package, including a 401(k) plan.
- Catered free lunch, unlimited snacks, and beverages.
- Wide range of opportunities for personal and professional development.
- Work in a future-oriented, innovative, and dynamic field.
Categories
ML EngineeringRobotics
About PlusAI
PlusAI develops AI-based virtual driver software that enables factory-built autonomous trucks for commercial freight operators. The company licenses its autonomy stack to truck OEMs and logistics partners; collaborators include TRATON’s Scania and MAN, Navistar’s International, Hyundai Motor Company, Iveco Group, Bosch, and DSV. Headquartered in Santa Clara with operations in the U.S. and Europe, PlusAI is privately held and in 2025 announced a planned public listing via a merger with Churchill Capital Corp IX.