Machine Learning Engineer, Data Curation - Autonomous Vehicles

Join NVIDIA’s Autonomous Vehicles Data Curation team to apply deep learning to search, curate, and evaluate datasets and ground truth that support self-driving car capabilities.

Responsibilities

• Keep current with deep learning research and innovations; implement and experiment with new ideas
• Work with researchers and engineers to convert research into reliable, industrial-strength machine learning models
• Take machine learning work from initial evaluation and experimentation through to shipping
• Define, collect, and curate training and evaluation datasets, emphasizing scenario coverage and label quality
• Develop vision-language models and embeddings to enable 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 MS (or equivalent experience) with 5+ years in a related technical field
• Relevant fields: Computer Science, Computer Engineering, or closely related technical discipline
• Proven experience building robust software
• Interest in Artificial Intelligence for robotics and autonomous navigation
• Motivation to learn new topics and solve hard problems
• Knowledge of mathematics
• Experience in Deep Learning / Machine Learning, including background in Computer Vision and/or multimodal learning
• Programming and debugging skills in C++ and/or Python
• Strong communication and analytical skills; ability to collaborate with multiple teams in a dynamic environment

Technologies

• PyTorch
• TensorFlow
• TensorRT
• Python
• C++

Benefits

• Equity and benefits
• Comprehensive benefits package

Ways to Stand Out

• Experience applying state-of-the-art AI methods to Computer Vision and Autonomous Vehicles problems
• Experience with Unsupervised or Self-supervised Learning and contrastive representation learning
• Experience with vision-language models, agentic search, or automated dataset quality evaluation

Location: Santa Clara, CA (onsite)

Salary: USD 152,000 - 287,500 per year