Machine Learning Engineer Intern - Planning

Reposted 29 Minutes Ago
Be an Early Applicant
Santa Clara, CA, USA
In-Office
Internship
Software
The Role
As a Machine Learning Engineer Intern, you'll modify models for ego-vehicle predictions, optimize loss functions, and evaluate trajectory diversity and accuracy.
Summary Generated by Built In
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.

In this role, the intern will expand the scene transformer model to provide multiple trajectories for ego in ambiguous cases, and explore longer horizon predictions using recursive and multi-step losses.
 
Area of work: 
Deep Learning Models, Planning, Prediction

Responsibilities:

  • Multimodal Architecture Design: Modify and expand an existing Scene Transformer framework to generate a diverse set of ego-vehicle trajectories for ambiguous scenarios.
  • Loss Function Engineering: Formulate and experiment with recursive, autoregressive, and multi-step losses to stabilize long-horizon trajectory rollouts.
  • Diversity & Feasibility Optimization: Mitigate mode collapse to ensure the generated paths are distinctly varied yet kinematically feasible.
  • Evaluation & Benchmarking: Design metrics to evaluate trajectory diversity, safety, and long-horizon accuracy against real-world driving logs.

Required Skills:

  • Expertise in trajectory prediction, behavior forecasting, and multi-agent motion modeling
  • Experience training large-scale models with self-supervision, contrastive learning, representation learning, or scene-level modeling
  • Practical experience with PyTorch, PyTorch Lightning or JAX.
  • Experience with uncertainty estimation (aleatoric, epistemic), multi-modal prediction, and probabilistic modeling
  • Understanding of AV stack components: localization, perception, tracking, prediction, and planning.
  • Strong debugging, experiment design, data analysis, and failure-mode investigation skills.

Your opportunities joining PlusAI
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.

Skills Required

  • Expertise in trajectory prediction
  • Experience training large-scale models
  • Practical experience with PyTorch or JAX
  • Experience with uncertainty estimation and multi-modal prediction
  • Understanding of AV stack components
  • Strong debugging and data analysis skills

Plus Compensation & Benefits Highlights

The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about Plus and has not been reviewed or approved by Plus.

  • Leave & Time Off Breadth Unlimited PTO in addition to company holidays and flexible work arrangements are offered, indicating broad time-off flexibility. This setup signals strong support for taking time away from work.
  • Healthcare Strength Tiered medical, dental, and vision options allow employees to select coverage that fits their needs. This breadth of core health coverage aligns with a comprehensive benefits approach.
  • Wellbeing & Lifestyle Benefits Daily catered lunches at key offices and company-sponsored professional development add meaningful day-to-day and growth-oriented perks. These offerings enhance overall wellbeing and workplace experience.

Plus Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Santa Clara, California
471 Employees
Year Founded: 2016

What We Do

Plus is a global provider of highly automated driving and fully autonomous driving solutions. Named by Forbes as one of America's Best Startup Employers and Fast Company as one of the World’s Most Innovative Companies, Plus's customers are already operating its product on the road today. Working with one of the largest companies in the U.S., vehicle manufacturers and others, Plus is making transportation safer and greener. Plus has received a number of industry awards and distinctions for its transformative technology and business momentum from Fast Company, Insider, Consumer Electronics Show, AUVSI, and others. For more information, visit www.plus.ai

Similar Jobs

Navan Logo Navan

Senior Product Manager

Fintech • Information Technology • Payments • Productivity • Software • Travel • Automation
Easy Apply
Hybrid
2 Locations
3300 Employees
115K-255K Annually

Parsec Automation Logo Parsec Automation

Customer Success Manager

Artificial Intelligence • Information Technology • Internet of Things • Software • Analytics • Automation • Manufacturing
Easy Apply
In-Office
Anaheim, CA, USA
99 Employees
250K-250K Annually

Braze Logo Braze

Solutions Engineer

Marketing Tech • Mobile • Software
Easy Apply
Hybrid
San Francisco, CA, USA
2000 Employees
86K-142K Annually
Hybrid
San Francisco, CA, USA
897 Employees
210K-240K Annually

Similar Companies Hiring

Hanover Park Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
42 Employees
Kepler  Thumbnail
Fintech • Software
New York, New York
6 Employees
Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account