Senior Machine Learning Engineer, Perception - Autonomous Driving

NVIDIA develops GPU-accelerated artificial intelligence technologies for computers, robots, and self-driving cars. The Senior Machine Learning Engineer will develop and productize perception solutions for autonomous driving, focusing on traffic signal understanding, deep learning models, data-driven development, and robust real-time deployment.

Responsibilities:

• Designing end2end solutions for Perception and AV stack to enable traffic signal detections across various driving environments from complex intersections to rural roads to highways
• Applied research and development of innovative 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 approaches to support diverse ODDs and Country/region expansion
• Drive and prioritize data-driven development by working with large data collection and labeling teams to bring in high value data to improve perception system accuracy. Efforts will include data collection prioritization and planning, labeling prioritization, labeling efficiency optimization, so that value of data is maximized
• Leverage data simulation and augmentation for solving extreme scenarios
• Productize the developed perception solutions by meeting product requirements for safety, latency, and SW robustness

Requirements:

• Minimum Requirement: PhD with 4+ years, MS with 6+ years, or BS (or equivalent experience) with 8+ years of relevant experience in Computer Science, Computer Engineering, or a related technical field
• Hands-on work experience in developing deep learning and algorithms to solve sophisticated real world problems, and proficiency in using deep learning frameworks (e.g., PyTorch)
• Experience in data-driven development and collaboration with data and ground truth teams
• Strong programming skills in python and/or C++
• Outstanding communication and teamwork skills as we work as a tightly-knit team, always discussing and learning from each other
• 2+ years of technical leadership demonstrating high technical and organizational complexity is a big plus
• Proven expertise in developing generalizable perception solutions for autonomous driving or robotics using deep learning with cameras
• Hands-on experience in developing and deploying DNN-based solutions to embedded platforms for real time applications
• Proven expertise in deep learning backed up by technical publications in leading conferences/journals
• Expertise with Visual Language Models, Transformers, BEV architectures, and modern traffic signal perception techniques. Experience in working on complex object detection and recognition problems is a big plus