Senior/Staff Machine Learning Engineer (Motion Planning)
PlusAI
- Santa Clara, United States
- Lead/Staff
- 4+ years
- Bachelor's degree
- Posted
Skills
Required
- Machine Learning
- planning / prediction / motion forecasting / trajectory generation
- autonomous
- ML
- robotics
- deep learning
- developing machine learning systems
- sequence modeling
- transformers
- diffusion models
- modern ML architectures
- modern C++
- production software development
Preferred
- autonomous driving
- TensorRT / ONNX Runtime / CUDA / ML inference optimization
- distributed training / cloud-based
- ML infrastructure
- autonomous vehicle planning / motion prediction
- validation methodologies
- Python programming
- PyTorch
Join our team as a Senior/Staff Machine Learning Engineer specializing in Motion Planning for autonomous vehicles. You will develop state-of-the-art machine learning models, design and implement model architectures, and own the end-to-end machine learning lifecycle. You will collaborate closely with various teams to deploy scalable machine learning solutions into production and ensure compliance with quality management standards.
Responsibilities
- Enjoy benefits such as unlimited PTO, flexible working arrangements, health benefits, equity options, professional development opportunities, and daily catered lunches.
- Développer des modèles d'apprentissage automatique de pointe pour la planification des véhicules autonomes en utilisant des données de capteurs.
- Concevoir et mettre en œuvre des architectures de modèles pour la génération de trajectoires, la planification du comportement et la prise de décision.
- Posséder le cycle de vie complet de l'apprentissage automatique, y compris la curation des données, l'ingénierie des fonctionnalités, l'expérimentation, la formation, l'évaluation, le déploiement.
- Strong ownership mindset with the ability to drive problems from investigation through implementation, validation, and deployment
- Experience designing datasets, experiments, validation methodologies, and metric-driven model evaluation
- Experience designing validation methodologies, automated testing, and monitoring to ensure correctness, safety, and production reliability
- Excellent debugging and analytical problem-solving skills, with the ability to investigate complex issues across datasets, model behavior, and production systems
- Strong Python programming skills and experience with modern deep learning frameworks such as PyTorch
- Experience analyzing edge cases, tracing failures to their root cause, and improving model robustness through systematic experimentation
Requirements
- 4+ years of experience developing machine learning systems for robotics, autonomous driving, or other real-time decision-making systems
- Strong software engineering skills with experience developing and maintaining production-quality software
- Excellent communication skills and experience collaborating across cross-functional engineering teams
- Strong understanding of deep learning, sequence modeling, transformers, diffusion models, or other modern ML architectures
- BS, MS, or PhD in Computer Science, Robotics, Machine Learning, or a related field
- Experience with planning, prediction, motion forecasting, or trajectory generation
- Working knowledge of modern C++ and production software development
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
Preferred
- Experience deploying machine learning models into production environments
- 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
- Candidates who stand out typically have one or more of the following:
- Deep expertise in autonomous vehicle planning or motion prediction
- Experience developing production ML systems for safety-critical applications
Pay
USD 130,000 - 220,000
jobapplier