The role is for a research engineer in the Autonomous Vehicles Research team at NVIDIA, focusing on software engineering and artificial intelligence. The team works on AI models for autonomous driving, including agent behavior models and end-to-end architectures. Candidates will engage in fundamental research while also impacting product development. Collaboration with various teams and domain scientists is essential to the role.

• 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.
• Strong programming skills in Python and a high-performance language such as C++.

• Develop large-scale supervised learning and reinforcement learning training frameworks for AVs.
• Optimize GPU and cluster utilization for efficient model training and fine-tuning.
• Implement scalable data loaders and preprocessors for multimodal datasets.
• Build and optimize simulation infrastructure to support the training of driving policies.
• Collaborate with researchers to integrate cutting-edge model architectures into training pipelines.

• Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.