Senior Machine Learning Engineer, Perception - Autonomous Driving

What you'll do

• Design, train, and optimize perception and sensor-fusion models; develop ML workflows for data, training, metrics, instrumentation, and reporting.
• Productize models and algorithms for NVIDIA DRIVE AV, including efficient C++ code.
• Collaborate across teams to deliver reliable autonomous-driving solutions and incorporate techniques that improve platform performance.

What they ask for

• MS or PhD in Computer Science, Engineering, or a related field, or equivalent experience.
• Strong C++ and Python programming and debugging skills, including experience with large, complex systems.
• Deep practical experience applying machine learning to lidar/camera perception and multi-sensor fusion in automotive or related fields.
• Experience with deep learning frameworks and ML mathematical foundations, and building and sustaining training and metric workflows for large-scale datasets.

Nice to have

• Track record shipping deep learning models for LiDAR/Camera and Multi-Sensor Fusion in production; experience delivering features into an autonomous vehicle or robotics product.
• Familiarity with Transformers or model optimization for real-time embedded/automotive inference, such as TensorRT.

Pay

184,000 USD - 287,500 USD (base salary, Level 4); 224,000 USD - 356,500 USD (base salary, Level 5)

Tools

C++, Python, PyTorch, TensorFlow, Transformers, TensorRT

AI focus

Build and productize deep-learning models and ML workflows for autonomous-vehicle perception, including LiDAR/camera perception, multi-sensor fusion, detection, classification, segmentation, and tracking.