Autonomous Vehicles Senior Research Engineer: ML at Scale

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

Job summary

NVIDIA Corporation is seeking a Senior Research Engineer for the Autonomous Vehicles Research team in Santa Clara, CA. You will drive large-scale training pipelines for multimodal AV models, optimize GPU usage, and build simulation infrastructure to scale driving policies across thousands of GPUs.

Ideal candidates have 10+ years in ML/AI infrastructure, proficiency in PyTorch/TensorFlow, CUDA, RL, and distributed systems, and a track record of publishable research.

Qualifications

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

Responsibilities

• Develop large-scale supervised learning and reinforcement learning training frameworks for AV multimodal foundation models on thousands of GPUs.
• Optimize GPU and cluster utilization for efficient 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) for driving policies training at scale.
• Collaborate with researchers to integrate state-of-the-art 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 and apply RL to fine-tune multimodal LLMs.

Skills

Python

C++

PyTorch

TensorFlow

JAX

CUDA

GPU acceleration

Reinforcement learning

Large-scale MLops

Education

Bachelor's degree in CS/Engineering/Robotics

Tools

PyTorch

TensorFlow

Kubernetes

SLURM

GPU clusters

Job description

NVIDIA Corporation is seeking a Senior Research Engineer for the Autonomous Vehicles Research team in Santa Clara, CA. You will drive large-scale training pipelines for multimodal AV models, optimize GPU usage, and build simulation infrastructure to scale driving policies across thousands of GPUs.

Ideal candidates have 10+ years in ML/AI infrastructure, proficiency in PyTorch/TensorFlow, CUDA, RL, and distributed systems, and a track record of publishable research.