As a Machine Learning Engineer Intern, you'll modify models for ego-vehicle predictions, optimize loss functions, and evaluate trajectory diversity and accuracy.
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
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.
Plus Santa Clara, California, USA Office
3315 Scott Blvd, Santa Clara, California, United States, 95054
Similar Jobs
Cloud • Software
The Accounting Manager will manage billing operations, revenue recognition, accounts receivable, sales tax compliance, and contribute to month-end close in a high-ownership role.
Top Skills:
ChargebeeDeelHubspotJustworksQuickbooks OnlineRampTabs
.jpg)
Marketing Tech • Mobile • Software
Sell Braze's customer engagement platform to large enterprises: build C-suite relationships, prospect and land net-new strategic accounts, apply consultative/value-based selling, manage and expand high-profile accounts, collaborate cross-functionally, and use Salesforce/Clari for pipeline management and forecasting.
Top Skills:
BrazeBrazeaiClariCRMSalesforce
Marketing Tech • Mobile • Software
As an Engagement Manager, you'll own AI deployment projects, leading workshops, driving process improvement, and engaging with clients to ensure program success.
Top Skills:
AIMachine Learning
What you need to know about the San Francisco Tech Scene
San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.
Key Facts About San Francisco Tech
- Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Google, Apple, Salesforce, Meta
- Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
- Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
- Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine
