Senior Machine Learning Engineer
Core
Design, train, and optimize machine learning models for LiDAR and camera perception in self-driving vehicles, shipping them as part of the NVIDIA DRIVE AV platform.
Role type
Senior Machine Learning Engineer (Autonomous Driving Perception)
Builds
Perception models and algorithms for the NVIDIA DRIVE AV platform
Domain
Autonomous driving, Computer Vision, LiDAR
Deliverable
production ML models
Required skills
C++, Python, Deep Learning frameworks (PyTorch, TensorFlow), LiDAR perception, Camera perception, Model optimization, Data pipeline development, Mathematical foundations of ML
Preferred skills
Transformers, TensorRT, Embedded systems inference, AV production stack experience
Technologies
PyTorch, TensorFlow, C++, Python, TensorRT, NVIDIA DRIVE AV
Responsibilities
Design and train ML models for road element detection, semantic segmentation, and tracking; Develop and coordinate end-to-end ML workflows including data pipelines and metrics; Productize models into efficient C++ code for the DRIVE AV platform; Collaborate with LiDAR/camera teams to deliver reliable solutions.
Seniority
Senior, hands-on IC
