Senior Machine Learning Engineer, Prediction - Autonomous Driving

at Nvidia

USD 224,000-356,500 per year

SENIOR

✅ On-site

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 @ 4 Communication @ 6 Deep Learning @ 4 GPU Leadership @ 4 Machine Learning @ 6 PyTorch @ 6 Python @ 7 Robotics @ 6 Technical Leadership @ 4

Details

NVIDIA is developing intelligent machines powered by artificial intelligence, GPU deep learning, and machine learning. The prediction team develops solutions for autonomous driving systems that produce multimodal, scene-consistent predictions for other road participants. This role focuses on improving the robustness, accuracy, and efficiency of prediction solutions for autonomous driving in diverse environments.

Responsibilities

Design end-to-end prediction and autonomous vehicle stack solutions using state-of-the-art machine learning techniques across driving environments, including complex intersections, rural curvy roads, and multilevel highways.
Conduct applied research and development of innovative deep learning models, including end-to-end models.
Develop generalizable approaches to support diverse operational design domains and country or regional expansion.
Drive and prioritize data-driven development in collaboration with large data collection and labeling teams.
Plan data collection and labeling priorities and optimize labeling efficiency to maximize the value of data for improving prediction system accuracy.
Leverage data simulation and augmentation to solve extreme scenarios.
Productize solutions while meeting requirements for safety, latency, and software robustness.

Requirements

PhD with 6 or more years of experience, master's degree with 10 or more years of experience, or bachelor's degree or equivalent experience with 12 or more years of meaningful experience in computer science, computer engineering, or a related technical field.
Hands-on experience developing deep learning models and algorithms to solve sophisticated real-world problems.
Proficiency with deep learning frameworks such as PyTorch.
Experience with data-driven development and collaboration with data and ground truth teams.
Strong programming skills in Python and/or C++.
Outstanding communication and teamwork skills.

Preferred Qualifications

Proven expertise developing generalizable machine learning solutions for autonomous driving or robotics.
Experience developing and deploying deep neural network-based solutions to embedded platforms for real-time applications.
Deep learning expertise supported by technical publications in leading conferences or journals.
Expertise with Transformers, bird's-eye-view architectures, and end-to-end machine learning systems.
More than two years of technical leadership experience involving significant technical and organizational challenges.

Compensation and Benefits

Base salary range: $224,000-$356,500 USD per year.
Eligibility for equity and benefits.

Applications will be accepted at least until October 20, 2026. NVIDIA uses AI tools in its recruiting processes and is committed to an inclusive work environment and equal opportunity employment.