Develop and deploy machine learning planning algorithms for autonomous trucks. Own the full ML lifecycle, including data curation, feature engineering, training, evaluation, deployment, monitoring, and improvement. Design trajectory and behavior-planning models, safety guardrails, validation pipelines, and performance metrics. Investigate failures and edge cases, optimize inference, and collaborate with perception, prediction, mapping, runtime, and systems teams to deliver reliable production systems.
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.
We’re looking for a machine learning engineer to train and deploy the latest generation of ML-based planning algorithms on the extensive data we collect every day across our autonomous trucking fleet.
Responsibilities
- Develop state-of-the-art machine learning models for autonomous vehicle planning using rich map, perception, routing, and contextual sensor data.
- Design and implement model architectures for trajectory generation, behavior planning, and decision making that balance accuracy, robustness, interpretability, and runtime efficiency.
- Own the end-to-end machine learning lifecycle, including data curation, feature engineering, experimentation, training, evaluation, deployment, monitoring, and continuous improvement.
- Design rigorous offline evaluation methodologies, validation pipelines, and metrics to measure planning quality, safety, robustness, and generalization.
- Analyze model behavior, investigate failure cases, and improve performance through systematic error analysis and targeted experimentation.
- Collaborate closely with runtime, perception, prediction, mapping, and systems teams to deploy scalable machine learning solutions into production.
- Design validation strategies and rule-based guardrails to ensure generated trajectories are feasible, safe, and compliant with traffic rules.
- Stay current with advances in machine learning, robotics, and autonomous driving, translating research innovations into production systems.
- Ensure technical work complies with the company's Quality Management System (QMS), customer requirements, regulatory standards, and internal engineering processes.
Required Skills
- BS, MS, or PhD in Computer Science, Robotics, Machine Learning, or a related field.
- 4+ years of experience developing machine learning systems for robotics, autonomous driving, or other real-time decision-making systems.
- Strong Python programming skills and experience with modern deep learning frameworks such as PyTorch.
- Strong understanding of deep learning, sequence modeling, transformers, diffusion models, or other modern ML architectures.
- Experience designing datasets, experiments, validation methodologies, and metric-driven model evaluation.
- Strong software engineering skills with experience developing and maintaining production-quality software.
- Excellent debugging and analytical problem-solving skills, with the ability to investigate complex issues across datasets, model behavior, and production systems.
- Experience analyzing edge cases, tracing failures to their root cause, and improving model robustness through systematic experimentation.
- Experience designing validation methodologies, automated testing, and monitoring to ensure correctness, safety, and production reliability.
- Strong ownership mindset with the ability to drive problems from investigation through implementation, validation, and deployment.
- Excellent communication skills and experience collaborating across cross-functional engineering teams.
Preferred Skills
- Experience with planning, prediction, motion forecasting, or trajectory generation.
- Experience deploying machine learning models into production environments.
- Working knowledge of modern C++ and production software development.
- Experience with TensorRT, ONNX Runtime, CUDA, or ML inference optimization.
- Experience with large-scale distributed training or cloud-based ML infrastructure.
- Publications or open-source contributions in machine learning, robotics, or autonomous driving.
- Deep expertise in autonomous vehicle planning or motion prediction.
- Experience developing production ML systems for safety-critical applications.
- Experience leading technical direction for large ML projects.
- Strong intuition for balancing model quality, robustness, latency, and deployment constraints.
- Demonstrated ability to solve ambiguous, cross-functional engineering problems involving machine learning, software systems, and autonomous driving.
Bonus Qualifications
Candidates who stand out typically have one or more of the following:
Candidates who stand out typically have one or more of the following:
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.
Similar Jobs
Software
Design, develop, and deploy reinforcement learning, behavior cloning, and imitation learning models for autonomous truck behavior planning. Own reinforcement learning pipelines from environment construction through training and modeling, contribute to distributed training infrastructure, analyze model performance and failure modes, and collaborate with simulation and planning teams to deploy models across driving scenarios.
Top Skills:
CC++GpudrivePufferdrivePythonPyTorch
Software
Design and optimize production machine-learning runtime pipelines for autonomous vehicle motion planning. Build high-performance C++ inference, feature extraction, trajectory generation, and orchestration components for embedded platforms. Develop validation, safety guardrails, fallback mechanisms, testing, monitoring, debugging, and observability tools. Investigate complex runtime failures and optimize latency, throughput, memory, and multithreaded performance while collaborating with machine-learning and systems engineering teams.
Top Skills:
C++C++17C++20Ci/CdCloud InfrastructureCudaEmbedded Gpu PlatformsOnnx RuntimePyTorchTensorrt
Artificial Intelligence • Software
Build and maintain knowledge and evaluation infrastructure for long-horizon AI agents. Develop production agents for complex customer use cases, analyze agent errors, and implement quality improvements. Build reliable backend systems supporting agent execution, evaluation, and monitoring, while partnering with senior engineers on platform reliability and quality. The role may also involve platform engineering, frontend work, and distributed systems.
Top Skills:
AIBackend SystemsDistributed SystemsLlms
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

