Machine Learning Engineer, Motion Planning & Prediction
Develops and deploys deep learning models for motion planning and behavioral prediction in autonomous vehicles using petabytes of driving data. Requires strong Python, PyTorch/TensorFlow expertise, full ML lifecycle experience, and C++ for real-time inference.
About the job
What You'll Do
- Design, train, and deploy state-of-the-art machine learning models for behavioral prediction and motion planning
- Develop robust data pipelines to process, clean, and label massive-scale vehicle sensor and simulation datasets
- Work with deep learning architectures such as transformers to model complex temporal interactions between traffic agents
- Establish and own the metrics for model performance, and create evaluation frameworks that correlate with on-road safety and performance
- Collaborate with software engineers to integrate and optimize trained models for real-time inference on the vehicles embedded hardware
- Stay current with the latest research in machine learning, imitation learning, and reinforcement learning, and apply novel techniques to our systems
What You'll Need
- Strong proficiency in Python and hands-on experience with modern deep learning frameworks (e.g., PyTorch, TensorFlow, or JAX)
- Solid understanding of machine learning fundamentals, including various neural network architectures, training methodologies, and evaluation techniques
- Experience with the full machine learning lifecycle, from data exploration and prototyping to deployment and monitoring
- Proficiency in C++ for writing high-performance model inference code
Nice to Have
- A strong track record in ML competitions (e.g., Kaggle) or contributions to major open-source ML projects
- Experience applying ML to problems in robotics, such as behavioral prediction, motion planning, or computer vision
- Experience with MLOps tools and platforms (e.g., MLflow, Kubeflow, Weights & Biases)
- Experience with large-scale distributed data processing and training frameworks (e.g., Spark, Ray)
- Publications in top-tier ML or robotics conferences (e.g., NeurIPS, ICML, CVPR, ICLR, CoRL, RSS)
Skills
Python, PyTorch, TensorFlow, JAX, C++, Transformers, MLOps, MLflow, Kubeflow, Weights & Biases, Spark, Ray
Similar jobs
ML Engineering jobs
Thinking Machines Lab
San Francisco, CA
Build and operate the engineering systems that support post-training research, including reinforcement learning infrastructure, sandboxed execution, data pipelines, and agent scaffolding. The role requires strong Python and systems engineering skills, project ownership, and a relevant bachelor’s degree or equivalent experience.
OpenAI
San Francisco, CA
Build research infrastructure and tooling that enables AI models to design silicon, including reinforcement learning environments, EDA integrations, evaluations, and experiment workflows. The role requires strong software engineering fundamentals and comfort working across research, tooling, and chip-design systems.
Rollstack
United StatesBuild production AI capabilities for automated slide and document generation, working across LLM applications, data analysis, and content generation. The role requires 3+ years in machine learning and NLP, advanced Python, and experience with LLM frameworks and production systems.
ClickUp
United States
Build and operate large-scale ranking and retrieval systems that power search relevance, including hybrid lexical/vector search, embeddings, query understanding, and permission-aware retrieval. Requires a bachelor's degree and 5+ years of ML engineering experience in ranking or information retrieval.
PathAI
Boston, MADevelop and deploy machine learning models for biomedical research and AI products, collaborating with scientific, engineering, and product teams. Requires an advanced quantitative degree, substantial ML experience, Python proficiency, and experience bringing models into production or research applications.