Senior Machine Learning Engineer, Perception - Autonomous Driving

NVIDIA is seeking a Senior Machine Learning Engineer specializing in perception for autonomous driving. The role focuses on developing and productizing traffic signal understanding solutions that do not rely on HD maps, with emphasis on end-to-end perception, deep learning model development, and performance improvements across robustness, accuracy, and efficiency.

Responsibilities

• Design end-to-end solutions spanning Perception and the AV stack to support traffic signal detections across varied driving environments, from complex intersections to rural roads and highways.
• Conduct applied research and development of deep learning models for traffic light detection, traffic sign recognition, road marking detection, construction object detection, text recognition, and other traffic signal tasks.
• Develop generalizable methods to support diverse operational design domains (ODDs) and enable country or region expansion.
• Lead data-driven development efforts by partnering with large data collection and labeling teams to prioritize collection and labeling, optimize labeling efficiency, and maximize the value of high-impact data to improve perception system accuracy.
• Use data simulation and augmentation to address extreme scenarios.
• Productize perception solutions by aligning with product requirements for safety, latency, and software robustness.

Requirements

• Minimum qualification: PhD with 4+ years, MS with 6+ years, or BS (or equivalent experience) with 8+ years in Computer Science, Computer Engineering, or a related technical field.
• 2+ years of technical leadership demonstrating high technical and organizational complexity is a plus.
• Hands-on experience building deep learning and algorithmic solutions for sophisticated real-world problems, including proficiency with deep learning frameworks such as PyTorch.
• Experience with data-driven development and collaboration with data and ground truth teams.
• Strong programming skills in Python and/or C++.
• Outstanding communication and teamwork skills in a tightly-knit team focused on ongoing discussion and learning.

Key Technologies

• PyTorch
• Python
• C++
• Deep learning frameworks
• Transformers
• BEV architectures
• Visual Language Models

Benefits

You will also be eligible for equity and benefits.

How to Stand Out

• Proven expertise developing generalizable perception solutions for autonomous driving or robotics using deep learning with cameras.
• Hands-on experience developing and deploying DNN-based solutions to embedded platforms for real-time applications.
• Proven deep learning expertise supported by technical publications in leading conferences and journals.
• Expertise with Visual Language Models, Transformers, BEV architectures, and modern traffic signal perception techniques, including strong performance on complex object detection and recognition problems.

Job Details

• Location: Santa Clara, CA (onsite)
• Salary Range: USD 184,000 - 356,500 per year
• Minimum Experience: 4 years
• Education: PhD