Senior Machine Learning Engineer

at Nvidia
USD 184,000-356,500 per year
SENIOR
✅ On-site

Tech Stack

AI Algorithms Communication @ 6 Computer Vision Data Pipelines Debugging @ 7 Deep Learning @ 7 Experimentation Machine Learning @ 7 PyTorch @ 4 Python @ 7 Robotics @ 4 TensorFlow @ 4 TensorRT @ 4

Details

NVIDIA's team is building the machine learning backbone for the perception component of NVIDIA DRIVE AV. The role focuses on solving challenging machine learning problems for self-driving cars under real-world conditions, including 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.
  • Take machine learning models and algorithms from evaluation and experimentation through productization 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, including 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 metric workflows for large-scale datasets.
  • Excellent communication and analytical skills, with a self-motivated drive to solve challenging problems.

Preferred Qualifications

  • A proven record of developing and shipping deep learning models for LiDAR, camera perception, or multi-sensor fusion in production environments.
  • 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 the use of TensorRT.

Benefits

  • Equity and employee benefits.
  • NVIDIA is committed to an inclusive work environment and is an equal opportunity employer.
  • Applications will be accepted at least until July 14, 2026.
  • This posting is for an existing vacancy. NVIDIA may use AI tools in its recruiting processes.

More jobs at Nvidia

Similar jobs