Senior Machine Learning Engineer, Perception - Autonomous Driving
Tech Stack
Tag name is followed by "@" symbol and proficiency level value.
About proficiency levels:
- 1-2 — basic awareness. Minimal hands-on experience, and a rudimentary understanding of the technology's purpose;
- 3-6 — daily use. Comfortable and regular usage, capable of handling common tasks and challenges related to the technology;
- 7-9 — you are an expert, you can teach others, you know all the pitfalls and tricks;
- 10 — exceptional knowledge, comprehensive understanding, and adeptness in all aspects of the technology, including advanced problem-solving. Think twice before claiming or demanding such level.
AI
Algorithms @ 7
Communication @ 6
Computer Vision
Data Pipelines
Debugging @ 7
Deep Learning @ 7
Experimentation
GPU @ 7
Machine Learning @ 7
PyTorch @ 4
Python @ 7
Robotics @ 4
TensorFlow @ 4
TensorRT @ 4
- 1-2 — basic awareness. Minimal hands-on experience, and a rudimentary understanding of the technology's purpose;
- 3-6 — daily use. Comfortable and regular usage, capable of handling common tasks and challenges related to the technology;
- 7-9 — you are an expert, you can teach others, you know all the pitfalls and tricks;
- 10 — exceptional knowledge, comprehensive understanding, and adeptness in all aspects of the technology, including advanced problem-solving. Think twice before claiming or demanding such level.
Details
Intelligent machines powered by artificial intelligence can learn, reason, and interact with people. NVIDIA's GPU runs deep learning algorithms that simulate human intelligence and power computers, robots, and self-driving vehicles that can perceive and understand the world.
The team is building the machine learning backbone of the perception component for NVIDIA DRIVE AV. The role focuses on solving challenging problems for self-driving cars involving computer vision, LiDAR and camera perception, multi-sensor fusion, and AI infrastructure.
Responsibilities
- Design, train, and optimize machine learning models for LiDAR and camera perception and multi-sensor fusion, including object detection and classification, image classification, semantic segmentation, and tracking.
- Develop and coordinate complete machine learning workflows, including data pipelines, model training, model metrics, continuous performance instrumentation, and reporting.
- Productize machine learning models and algorithms from initial evaluation and experimentation through deployment on the NVIDIA DRIVE AV platform.
- Develop highly efficient product code in C++.
- Track the latest developments in machine learning and incorporate techniques that improve platform performance.
- Collaborate with LiDAR and camera teams, developers, engineers, and managers to turn complex ideas into reliable autonomous-driving solutions.
Requirements
- MS or PhD in Computer Science, Engineering, or a related field, or equivalent experience.
- 8+ years of relevant industry experience applying machine learning to real-world problems.
- Strong C++ and Python programming and debugging skills, with experience developing for large, complex systems.
- Deep practical experience applying machine learning to LiDAR and camera perception and multi-sensor fusion in automotive or related fields.
- Experience with deep learning frameworks such as PyTorch and TensorFlow.
- Strong understanding of the mathematical foundations of machine learning.
- Experience building and sustaining training and essential metric workflows for large-scale datasets.
- Excellent communication and analytical skills, with a self-motivated drive to solve challenging problems.
Preferred Qualifications
- A proven track record of developing and shipping deep learning models for LiDAR, camera perception, and multi-sensor fusion in a production environment.
- Familiarity with modern network architectures such as Transformers and their application to visual recognition tasks.
- Experience delivering machine learning features and models into a production autonomous vehicle stack or related robotics product.
- Experience optimizing models for real-time inference on embedded or automotive platforms, including TensorRT.
Compensation and Benefits
The base salary depends on location, experience, and the pay of employees in similar positions. The base salary range is USD 184,000–287,500 for Level 4 and USD 224,000–356,500 for Level 5. The position is also eligible for equity and benefits.
Applications will be accepted at least until October 10, 2026. This posting is for an existing vacancy.
NVIDIA uses AI tools in its recruiting processes. NVIDIA is committed to fostering an inclusive work environment and is an equal opportunity employer. The company does not discriminate on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status, or any other characteristic protected by law.