Senior Machine Learning Engineer, Prediction - Autonomous Driving
Core
Develop and productize prediction modules for NVIDIA's autonomous driving solutions, producing multi-modal scene consistent predictions for road participants.
Role type
Senior IC machine learning engineer (autonomous driving prediction)
Builds
End-to-end prediction solutions for AV stacks, deep learning models, and data-driven development pipelines.
Domain
Autonomous driving, Computer Vision, Deep Learning
Deliverable
production ML models
Required skills
Deep learning model development, PyTorch, Python, C++, data-driven development, data simulation and augmentation, embedded platform deployment, Transformers, BEV architectures, E2E ML systems
Preferred skills
Technical publications in leading conferences, technical leadership experience
Technologies
PyTorch, Python, C++, Transformers, BEV architectures
Responsibilities
Design end-to-end prediction solutions for diverse driving environments, develop generalizable approaches for ODD expansion, prioritize and optimize data collection and labeling, leverage data simulation for extreme scenarios, productize solutions for safety and latency requirements
Seniority
Senior, hands-on IC