Drimva

NVIDIA

Senior Machine Learning Engineer, Prediction - Autonomous Driving

$224K–$356K

US, CA, Santa Clara +1 · Senior · Full-time · Posted Oct 29, 2026

Location
US, CA, Santa Clara
US, CA, Remote
Experience
PhD with 6+ years, MS with 10+ years, or BS (or equivalent experience) with 12+ years; over 2 years of technical leadership is advantageous.
Education
PhD, MS, or BS (or equivalent experience) in Computer Science, Computer Engineering, or a related technical field.

About the job

Develops and productizes machine-learning prediction systems for autonomous driving, improving their accuracy, robustness, and efficiency across diverse driving environments.

What you'll do

  • Design end-to-end prediction and autonomous-driving solutions using machine-learning techniques; develop deep-learning models, including end-to-end models.
  • Develop generalizable approaches for diverse operational design domains and regions, using data, simulation, and augmentation to improve prediction.
  • Productize solutions to meet safety, latency, and software-robustness requirements; coordinate data collection and labeling priorities with data teams.

What they ask for

  • PhD with 6+ years, MS with 10+ years, or BS (or equivalent experience) with 12+ years in Computer Science, Computer Engineering, or a related technical field.
  • Hands-on experience developing deep learning and algorithms for sophisticated real-world problems, and proficiency with deep-learning frameworks.
  • Experience in data-driven development and collaboration with data and ground-truth teams.
  • Strong programming skills in python and/or C++.

Nice to have

  • Expertise in generalizable ML for autonomous driving or robotics, or deploying DNN-based solutions on embedded platforms for real-time applications.
  • Expertise with Transformers, BEV architectures, or E2E ML systems; technical publications and over 2 years of technical leadership are also advantageous.

Pay

224,000 USD - 356,500 USD (base salary)

Tools

PyTorch, python, C++, Transformers, BEV architectures, E2E ML systems

AI focus

Applied research, development, and productization of deep-learning prediction models for autonomous driving, including end-to-end models.

Model-written summary · the original posting is authoritative

First seen Oct 17, 2026 · Last verified Nov 4, 2026 (UTC)