← All roles

Senior Machine Learning Engineer, Prediction - Autonomous Driving

NVIDIA· Engineer, Sys SW
Apply on company site
Hybrid (Santa Clara, CA), Remote$224k - $356.5kfull-timePosted 4d ago

About the role

Senior engineer to develop and productize NVIDIA’s autonomous driving prediction solutions, focusing on multi-modal scene-consistent prediction for road participants, with emphasis on robustness, accuracy, and efficiency across diverse driving environments.

Responsibilities

  • Designing end-to-end solutions for Prediction and AV stack using state of the art ML techniques to enable agents prediction across various driving environments from complex intersections to rural curvy roads to multi-level highways.
  • Applied research and development of innovative deep learning models, including end-to-end models
  • 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 prediction 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 solutions by meeting product requirements for safety, latency, and SW robustness.

Qualifications

  • Minimum Requirement: PhD with 6+ years, MS with 10+ years, or BS (or equivalent experience) with 12+ years of meaningful 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.

Skills

Required:PhD/MS/BS with relevant yearsDeep learningPyTorchPython/C++Data-driven developmentCommunication & teamwork

Benefits

  • equity
  • benefits
Last updated Oct 23, 2026. Interested? Apply on the company site.

About NVIDIA

World Leader in AI Computing

NVIDIA invents the GPU and drives advances in AI, HPC, gaming, creative design, autonomous vehicles, and robotics.

AIautonomous vehiclesgaminggpugraphics cardhigh performance computingliquid-ai-recruiting-targetllm
Company profileWebsite

More open roles at NVIDIA