Senior/Staff Machine Learning Engineer, Motion Planning

PlusAI, Inc.; Santa Clara (CA); Hybrid; USD 180,000 - 240,000; Full time

Benefits offered by this job

Catered lunch

Unlimited snacks and beverages

401(k) plan

Job summary

PlusAI, Inc. is seeking a Senior/Staff Machine Learning Engineer to train and deploy ML-based planning algorithms for autonomous trucks. You will work with rich map, perception, routing and sensor data to build trajectory generation and behavior planning models.

You will own data curation, experimentation, training, deployment, monitoring and continuous improvement in a production environment. You will collaborate with runtime, perception, mapping and systems teams to ensure safety, robustness

Qualifications

• BS, MS, or PhD in Computer Science, Robotics, ML or related field.
• 4+ years building ML systems for robotics or real-time decision making.
• Strong Python and PyTorch experience.
• Expertise in deep learning, sequence models, transformers or diffusion models.
• Experience designing datasets, experiments and metric-driven evals.
• Production-grade software engineering experience.
• Strong debugging and analytical skills for complex data and models.
• Experience validating models and improving robustness via experiments.
• Ability to drive projects end-to-end with cross-functional teams.
• Excellent communication across engineering teams.

Responsibilities

• Develop ML models for autonomous vehicle planning using map, perception, routing and sensor data.
• Design architectures for trajectory generation and behavior planning.
• Own end-to-end ML lifecycle: data curation, feature engineering, training and deployment.
• Create offline eval strategies and metrics for safety and robustness.
• Analyze model behavior and improve performance via targeted experiments.
• Collaborate with runtime, perception, mapping and systems teams to deploy ML in production.
• Design validation strategies to ensure safe, feasible trajectories.
• Stay current with ML and autonomy advances and translate research into production.
• Ensure work complies with QMS, customer requirements and standards.

Skills

Python programming

Deep learning

PyTorch

ML systems

Transformers

Experimentation

Production software

Cross-functional work

Debugging

System optimization

Education

CS/Robotics degree

Tools

TensorRT

ONNX Runtime

CUDA

C++ (production)

Cloud ML infrastructure

Job description

Senior/Staff Machine Learning Engineer, Motion Planning

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 amachine learning engineer to train and deploy the latest generation ofML-basedplanning algorithmson the extensive data we collect every dayacrossour 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.

Bonus Qualifications

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

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.

Whether you need an AI driver for your next vehicle program or a platform to build autonomy you control, PlusAI meets you where you are.