Senior AI & Simulation Engineer for Autonomous Driving RL

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

Job summary

NVIDIA Corporation in Santa Clara, California seeks exceptional Senior Machine Learning and Simulation Engineers to join the Autonomous Vehicles Simulation team. You will design and optimize large-scale RL training frameworks for multi-modal AV models and build scalable simulation pipelines for driving policies.

The role requires deep expertise in RL, AI, and simulation, with strong C++ and Python programming, experience on GPU clusters, and collaboration with AV Platform, Product, and Research

Qualifications

• Bachelor's degree in Computer Science, Robotics, Engineering, or a related field.
• 12+ years of relevant professional experience encompassing large-scale ML training, AV systems, simulation, and AI infrastructure development.
• Deep proficiency in RL algorithms, such as PPO and GRPO, including practical experience with hyperparameter tuning and reward function design.
• Exceptional programming skills in C++ and Python, vital for developing efficient systems and data pipelines.
• Extensive experience with large-scale GPU clusters, HPC environments, and job scheduling/orchestration tools (e.g., Kubernetes, SLURM).

Responsibilities

• Lead the design and development of large-scale RL training frameworks to accelerate the development of multi-modal AV foundation models.
• Design, build, and optimize simulation and data processing pipelines to enable scalable training of driving policies.
• Focus on measuring and enhancing simulation quality and refining the reward function for RL training.
• Ensure the reliability and performance of training workflows on large GPU clusters through the development of robust monitoring and debugging tools.
• Partner with researchers to integrate state-of-the-art model architectures into efficient and scalable training pipelines.

Skills

C++

Python

Reinforcement learning

PPO/GRPO

Large-scale ML training

GPU clusters

HPC environments

Kubernetes

SLURM

Education

Bachelor's degree in Computer Science/Engineering/Robotics

Tools

Kubernetes

SLURM

Job description

NVIDIA Corporation in Santa Clara, California seeks exceptional Senior Machine Learning and Simulation Engineers to join the Autonomous Vehicles Simulation team. You will design and optimize large-scale RL training frameworks for multi-modal AV models and build scalable simulation pipelines for driving policies.

The role requires deep expertise in RL, AI, and simulation, with strong C++ and Python programming, experience on GPU clusters, and collaboration with AV Platform, Product, and Research