Senior Research Engineer - Autonomous Vehicles

NVIDIA Corporation; Santa Clara (CA); On-site; USD 184,000 - 357,000; Full time

Benefits offered by this job

Equity and Benefits

Job summary

NVIDIA is seeking top research engineers for the Autonomous Vehicles Research team. Join a collaborative, publish-forward group focused on AI for autonomous driving, including foundation models, multimodal data, and sim-to-real pipelines.

The role blends fundamental research freedom with product collaboration across CUDA, simulation, robotics, and AI product teams, leveraging massive GPU infrastructure and HPC tooling.

Qualifications

• Bachelor's degree or equivalent in Computer Science, Engineering, or related field.
• 10+ years of full-time industry experience in large-scale MLOps and AI infrastructure.
• Proven experience designing and optimizing distributed training systems with PyTorch, JAX, or TensorFlow.
• Deep familiarity with reinforcement learning algorithms like PPO, SAC, or Q-learning, including hyperparameter tuning.
• Familiarity with policy learning techniques like reward shaping, domain randomization, curriculum learning.
• Deep understanding of GPU acceleration, CUDA programming, and cluster management tools like Kubernetes.

Responsibilities

• Develop large-scale supervised learning and reinforcement learning training frameworks to support multimodal foundation models for AVs on thousands of GPUs.
• Optimize GPU and cluster utilization for training and fine-tuning on massive datasets.
• Implement scalable data loaders and preprocessors for multimodal data (videos, text, sensor data).
• Build and optimize simulation infrastructure (GPU-accelerated simulators) for driving policies at scale.
• Collaborate with researchers to integrate cutting-edge model architectures into scalable pipelines.
• Develop sim-to-real transfer pipelines and deploy to real-world cars with AV product teams.
• Propose scalable solutions that combine LLMs with policy learning.
• Apply reinforcement learning to fine-tune multimodal LLMs.
• Develop robust monitoring and debugging tools for training workflows on large GPU clusters.

Skills

Python

C++

PyTorch

JAX

TensorFlow

CUDA

Distributed training

Reinforcement learning

GPU acceleration

Kubernetes

SLURM

High-performance computing

Education

Bachelor's degree or equivalent experience

Tools

PyTorch

JAX

TensorFlow

CUDA

Kubernetes

SLURM

Job description

We are recruiting top research engineers in the Autonomous Vehicles Research team at NVIDIA with strong expertise in software engineering and in artificial intelligence topics, such as deep learning, reinforcement learning, and generative modeling. You must have strong programming skills, a solid track record of training deep learning models at scale, and a good mathematical foundation to analyze new AI algorithms. We focus on AI models for autonomous driving such as agent behavior models, end-to-end AV architectures, AI safety, closed-loop training approaches, and AV foundation models (VLMs, reasoning models, etc.). We will be publishing at top venues and working with the broader scientific community. Communicating with different teams and domain scientists in different areas is essential. The position will aid fundamental research with the freedom and bandwidth to conduct ground-breaking publishable research. At the same time, you will also have the opportunity to impact products and collaborate with teams that focus on AI products based on CUDA, physically-based simulation, graphics, natural language processing, autonomous driving, HW optimization, robotics, healthcare, and many more. NVIDIA has an open and nurturing atmosphere for research that encourages collaboration.

What you will be doing:

• Develop large-scale supervised learning and reinforcement learning training frameworks to support multi-modal foundation models for AVs capable of running on thousands of GPUs;
• Optimize GPU and cluster utilization for efficient model training and fine-tuning on massive datasets;
• Implement scalable data loaders and preprocessors tailored for multimodal datasets, including videos, text, and sensor data;
• Build and optimize simulation infrastructure (based on GPU-accelerated simulators) to support the training of driving policies for AVs at scale;
• Collaborate with researchers to integrate cutting-edge model architectures into scalable training pipelines;
• Develop sim-to-real transfer pipelines and work closely with the AV product team to deploy to real-world cars;
• Propose scalable solutions that combine LLMs with policy learning;
• Apply reinforcement learning to finetune multimodal LLMs;
• Develop robust monitoring and debugging tools to ensure the reliability and performance of training workflows on large GPU clusters.

What we need to see:

• Bachelor's degree in Computer Science, Robotics, Engineering, or a related field or equivalent experience.
• 10+ years of full-time industry experience in large-scale MLOps and AI infrastructure.
• Proven experience designing and optimizing distributed training systems with frameworks like PyTorch, JAX, or TensorFlow.
• Deep familiarity with reinforcement learning algorithms like PPO, SAC, or Q-learning, including experience tuning hyperparameters and reward functions.
• Familiarity with common policy learning techniques like reward shaping, domain randomization, curriculum learning.
• Deep understanding of GPU acceleration, CUDA programming, and cluster management tools like Kubernetes.
• Strong programming skills in Python and a high-performance language such as C++ for efficient system development.
• Strong experience with large-scale GPU clusters, HPC environments, and job scheduling/orchestration tools (e.g., SLURM, Kubernetes).

Compensation and Benefits

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

• The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5.
• You will also be eligible for equity and benefits.

Equal Opportunity Employer

NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.

Learn more about NVIDIA.