jobapplier

Senior/Staff Machine Learning Engineer (Motion Planning)

PlusAI

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