Reinforcement Learning Planning Research Intern

Plus 2; Santa Clara (CA); On-site; USD 53,568 - 71,424; Full time

Benefits offered by this job

Catered free lunch

Unlimited snacks and beverages

401(k) plan

Job summary

PlusAI, a Silicon Valley–based Physical AI company, invites summer interns to own the development of a Safety-Critical Trajectory Correction (STC) module. You will design, train, and validate this architecture using Deep Reinforcement Learning to provide a continuous safety barrier for our autonomous truck fleet.

The role emphasizes conducting research in reinforcement learning, developing deep learning techniques, and integrating results into the production perception/AV stack, with

Qualifications

• Pursuing MS or PhD in CS, EE, mathematics, statistics or related field.
• Thorough understanding of reinforcement learning.
• 1-2 years experience with implementing and training models in at least one deep learning framework (PyTorch, Tensorflow, Jax).
• Past experiences in design, implementation and training of deep reinforcement learning models.
• Past experiences in projects related to autonomous driving.

Responsibilities

• Conduct groundbreaking research with the potential to strongly impact PlusAI’s autonomous driving products.
• Key focus area for this internship will be reinforcement learning to generate safe trajectories for autonomous driving.
• Develop and benchmark cutting-edge techniques in deep learning.
• Collaborate with team members to optimize and seamlessly integrate developed techniques into the production perception/AV stack.

Skills

Reinforcement learning

Deep learning

PyTorch

TensorFlow

JAX

Education

MS or PhD in CS, EE, mathematics, statistics or related field

Tools

PyTorch

TensorFlow

JAX

Job description

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.

Autonomous vehicles (AVs) need to navigate complex environments not just smoothly, but with absolute safety. While traditional planners handle standard driving behavior, edge-case scenarios and sudden environmental changes require an unyielding safety net.

This summer, you will own the development of a Safety-Critical Trajectory Correction (STC) module. Acting as a real-time safety overlay, the STC will function as a fallback mechanism that intercepts and minimally perturbs intended trajectories when collision risks are detected. You will design, train, and validate this architecture using Deep Reinforcement Learning to provide a continuous, constrained safety barrier for our vehicle fleet.

Responsibilities

• Conduct groundbreaking research with the potential to strongly impact Plus’s autonomous driving products, leading to publishable results. Key focus area for this internship will be reinforcement learning to generate safe trajectories for autonomous driving.
• Develop and benchmark cutting-edge techniques in deep learning.
• Collaborate with team members to optimize and seamlessly integrate developed techniques into the production perception/AV stack.

Qualifications

• Pursuing MS or PhD in CS, EE, mathematics, statistics or related field.
• Thorough understanding of reinforcement learning.
• 1-2 years experience with implementing and training models in at least one deep learning framework (PyTorch, Tensorflow, Jax).
• Past experiences in design, implementation and training of deep reinforcement learning models.
• Past experiences in projects related to autonomous driving.

Benefits

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