Senior Machine Learning Engineer

NVIDIA Gruppe; Santa Clara (CA); On-site; USD 210,000 - 315,000; Full time

Benefits offered by this job

Equity

Comprehensive benefits

Job summary

NVIDIA in Santa Clara, CA is seeking a machine learning engineer with a strong background in computer vision, LiDAR and AI infrastructure to advance the DRIVE AV perception stack. You will design, train, and optimize ML models and collaborate with cross-functional teams to ship production-ready components.

The role requires 6+ years in ML for automotive perception, expert C++/Python, and experience with PyTorch or TensorFlow.

Qualifications

• 6+ years of relevant proven industry experience applying machine learning to address real-world problems.
• BS or MS in Computer Science, Engineering, or a related field, or equivalent experience.
• Strong C++ and Python programming and debugging skills with experience in developing for large, complex systems.
• Deep practical experience applying machine learning to LiDAR/camera perception in automotive or related fields.
• Experience with deep learning frameworks (e.g., PyTorch, TensorFlow) and a strong understanding of the mathematical foundations of ML.
• Building and sustaining training and essential metric workflows for large-scale datasets.

Responsibilities

• Model Development: Design, train, and optimize ML models for LiDAR perception (road element detection, semantic segmentation, tracking).
• Develop ML workflows: data pipelines, model training, metrics, and reporting.
• Productization: Ship ML models as part of NVIDIA DRIVE AV via production-grade code in C++.
• Innovation: Track and apply new ML techniques to improve platform performance.
• Collaborate with LiDAR/camera teams and engineers to turn complex ideas into reliable autonomous driving solutions.

Skills

C++

Python

Machine learning

LiDAR perception

PyTorch

TensorFlow

Education

BS/MS in Computer Science or related field

Tools

CUDA

Linux

TensorRT

ROS

Job description

Overview

Intelligent machines powered by Artificial Intelligence computers that can learn, reason, and interact with people are no longer science fiction. Today, a self-driving vehicle powered by AI can meander through a country road at night and find its way. NVIDIA's GPU runs Deep Learning algorithms, simulating human intelligence, and acts as the brain of computers, robots, and self-driving vehicles that can perceive and understand the world. Our team is building the machine learning backbone of the Perception component for NVIDIA DRIVE AV. We are seeking the best Machine Learning Engineers with a background in computer vision, LiDAR & camera perception, and AI infrastructure who are passionate about solving the hardest problems for self-driving cars. Are you interested in inventing human‑level AI for perception in the unconstrained world under any conditions? If so, join us!

What You’ll Be Doing

• Model Development: Design, train, and optimize innovative machine learning models for LiDAR perception (e.g., road element detection, semantic segmentation, tracking).
• Develop and coordinate entire ML workflows, covering data pipelines, model training, model metrics, continuous performance instrumentation, and reporting.
• Productization: Take ML models and algorithms from initial evaluation and experimentation all the way to shipping them as part of the NVIDIA DRIVE AV platform, developing highly efficient product code in C++.
• Innovation: Keep track of the latest developments in machine learning, and incorporate techniques that improve platform performance.
• Collaborate with LiDAR/camera teams, developers, engineers, and managers to turn complex ideas into reliable solutions for autonomous driving.

What We Need to See

• BS or MS in Computer Science, Engineering, or a related field, or equivalent experience.
• 6+ years of relevant proven industry experience applying machine learning to address real‑world problems.
• Strong C++ and Python programming and debugging skills with experience in developing for large, complex systems.
• Deep practical experience applying machine learning to lidar/camera perception in automotive or related fields.
• Experience with deep learning frameworks (e.g., PyTorch, TensorFlow) and a strong understanding of the mathematical foundations of ML.
• Building and sustaining training and essential metric workflows for large‑scale datasets.
• Excellent communication and analytical skills. Self‑motivated drive to solve hard problems.

Ways to Stand Out From the Crowd

• LiDAR or Camera Perception Experience: Proven track record of developing and shipping deep learning models for LiDAR/Camera in a production environment.
• Advanced Model Knowledge: Familiarity with modern network architectures like Transformers and their application to visual recognition tasks.
• AV Production Experience: A history of delivering ML features and models into a production autonomous vehicle stack or a related robotics product.
• Performance Optimization: Experience with model optimization for real‑time inference on embedded or automotive platforms (e.g., using TensorRT).

Compensation and Benefits

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184000 USD - 287500 USD for Level 4, and 224000 USD - 356500 USD for Level 5. You will also be eligible for equity and benefits.

Application Deadline

Applications for this job will be accepted at least until July 9, 2026.

NVIDIA is committed to fostering an inclusive work environment and is proud to be an equal opportunity employer. We do not discriminate (including in our hiring and promotion practices) 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.