Machine Learning Engineer, Data Curation - Autonomous Vehicles
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 @ 3
Communication @ 6
Computer Vision @ 3
Debugging @ 3
Deep Learning @ 3
Experimentation
GPU
Machine Learning @ 3
PyTorch
Python @ 3
Robotics @ 3
TensorFlow
TensorRT
- 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.
Details
We are looking for a Machine Learning Engineer for our Autonomous Vehicles Data Curation team. The role involves applying state-of-the-art techniques to search, curate, and evaluate high-quality datasets and ground truth for autonomous vehicles, a critical aspect of NVIDIA's next-generation products. You will work with researchers and engineers in deep learning and computer vision to deliver impact to customers around the world.
NVIDIA GPUs are used broadly for deep learning. The team is training deep neural networks and developing data search and curation capabilities for NVIDIA's Autonomous Vehicles effort, with the goal of enabling autonomous driving. The role focuses on building datasets and tools that support perception, prediction, planning, and control for self-driving cars.
Responsibilities
- Stay up to date with the latest research and innovations in deep learning, and implement and experiment with new ideas.
- Foster collaborative partnerships with researchers and engineers, transforming innovative research into robust, industrial-strength machine learning models.
- Take approaches from initial evaluation and experimentation through to shipping.
- Define, collect, and curate training and evaluation datasets, with a focus on scenario coverage and label quality.
- Develop vision-language models and embeddings for scenario search, classification, and automated data curation.
- Build training pipelines and scalable inference systems using PyTorch, TensorFlow, TensorRT, Python, and C++.
Requirements
- PhD with 1+ year of relevant experience, or a master's degree or equivalent experience with 5+ years of relevant experience in Computer Science, Computer Engineering, or a related technical field.
- Proven experience building robust software.
- Passion for artificial intelligence applied to robotics and autonomous navigation.
- Strong learning, problem-solving, mathematical, communication, and analytical skills.
- Experience in deep learning or machine learning, with a background in computer vision and/or multimodal learning.
- Programming and debugging skills in C++ and/or Python.
- Ability to work with multiple teams in a dynamic environment.
Preferred Qualifications
- Experience applying the latest AI methods to computer vision and autonomous vehicle problems.
- Experience with unsupervised or self-supervised learning and contrastive representation learning.
- Experience with vision-language models, agentic search, or automated dataset quality evaluation.
Benefits
NVIDIA offers competitive salaries, equity, and a comprehensive benefits package. The position's expected base salary ranges are USD 152,000–241,500 for Level 3 and USD 184,000–287,500 for Level 4. The base salary is determined by location, experience, and other objective job-related factors.
Applications will be accepted at least until November 11, 2026. This posting is for an existing vacancy. NVIDIA uses AI tools in its recruiting processes and is an equal opportunity employer.