Machine Learning Engineer, Behavior Planning & Prediction

RESPONSIBILITIES

• Support the design and the development of advanced machine learning models in the behavior space specifically tailored for autonomous vehicles utilizing deep learning and large-scale data analysis.
• Drive projects through the execution phase, requiring minimal guidance by a senior engineer.
• Deploy scalable and efficient ML models on our autonomous vehicle platform.
• Integrate modern technologies with rigorous safety standards while maintaining cost efficiency.
• Significantly help developing needed components for end-to-end ML training and deployment, from data strategy to optimization and validation.
• Write high-quality code while being rigorous with Machine Learning experimentation.
• Work in a high-velocity environment and employ agile development practices.
• Collaborate closely with teams such as Perception, Simulation, Infrastructure, Data and Tooling to drive unified solutions.

MINIMUM QUALIFICATIONS

• B.S, M.S in Machine Learning, Computer Science, Robotics or related quantitative fields, or equivalent industry experience.
• 2+ years of  strong experience with Python, any major deep learning framework, and software engineering best practices
• Comfortable in writing C++ code to help integrate with our autonomous vehicle platform.
• 2+ years of experience with deep learning approaches such as supervised/unsupervised learning, transfer learning, multi-task learning, and/or deep reinforcement learning.
• Solid experience with learning-based planning approaches like imitation learning, reinforcement learning and state-of-the-art techniques for sequential modeling like Transformer architectures
• 2+ years of experience covering machine learning workflows, data sampling and curation, pre-processing, model training, ablation studies, evaluation, deployment, and inference optimization.
• Passionate about self driving car technology and its potential for humanity.
• Strong communication skills with the ability to communicate concepts clearly and precisely.

NICE TO HAVES

• Hands-on experience with autonomous driving systems
• Experience with model deployment with NVIDIA stack (e.g. ONNX graphs, TensorRT, profiling)
• Familiarity with recent breakthroughs in ML (e.g. foundation models, pre-training and efficient fine-tuning, multimodal Transformer architectures)
• Knowledge of autonomous driving large scale data curation pipelines