Senior Research Engineer for Reinforcement Learning

NVIDIA is the world leader in accelerated computing, pioneering solutions in AI and digital twins.
$148,000 - $287,500
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI · Robotics

Description For Senior Research Engineer for Reinforcement Learning

NVIDIA is seeking a senior or principal engineer specializing in large-scale reinforcement learning and policy learning for their Generalist Embodied Agent Research (GEAR) group. This role is part of Project GR00T, NVIDIA's ambitious initiative to develop foundation models and technology for humanoid robots. The position offers the opportunity to work with a distinguished research team known for influential work in multimodal foundation models, robot learning, embodied AI, and physics simulation.

The role combines cutting-edge research with practical engineering challenges, requiring expertise in both reinforcement learning and large-scale distributed systems. Key responsibilities include developing training frameworks that can scale to thousands of GPUs, building simulation infrastructure for robot training, and implementing sim-to-real transfer solutions. The successful candidate will work directly with the robotics team to deploy solutions on physical robots and contribute to innovative projects combining LLMs with policy learning.

This position at NVIDIA, a global leader in accelerated computing, offers competitive compensation including a base salary range of $148,000-$287,500 USD, plus equity and benefits. The company's work in AI and digital twins is transforming major industries, making this an opportunity to contribute to groundbreaking technological advances. The role is based in Santa Clara, CA, and requires 5+ years of industry experience, strong engineering skills, and deep expertise in reinforcement learning.

The ideal candidate will have advanced degrees in relevant fields, experience with robot policy transfer, and a track record of contributions to open-source frameworks or research publications. This position offers the chance to be at the forefront of developing general-purpose robots and large-scale foundation models while working with some of the most innovative minds in the industry.

Last updated 5 minutes ago

Responsibilities For Senior Research Engineer for Reinforcement Learning

  • Develop large-scale reinforcement learning training framework for thousands of GPUs
  • Build and optimize simulation infrastructure for robot training
  • Develop sim-to-real transfer pipelines
  • Work with robotics team to deploy to physical robots
  • Propose scalable solutions combining LLMs with policy learning
  • Apply reinforcement learning to finetune multimodal LLMs

Requirements For Senior Research Engineer for Reinforcement Learning

Python
  • Bachelor's degree or above in Computer Science, Robotics, Engineering, or related field
  • 5+ years of industry experience in large-scale deep learning or MLOps
  • Exceptional engineering skills in building scalable distributed GPU training frameworks
  • Proficiency in Python and hands-on experience with PyTorch, JAX, or Tensorflow
  • Deep familiarity with reinforcement learning algorithms
  • Experience with policy learning techniques
  • Strong experience with large-scale GPU clusters and HPC environments

Benefits For Senior Research Engineer for Reinforcement Learning

Equity
Medical Insurance
  • Equity
  • Medical Insurance

Interested in this job?

Jobs Related To NVIDIA Senior Research Engineer for Reinforcement Learning

Senior Compiler Engineer, Software - Deep Learning Accelerator

Senior Compiler Engineer position at NVIDIA focusing on Deep Learning Accelerator development, requiring 7+ years of experience in compiler or deep learning software projects.

Full Stack Developer, AI and LLM

Senior Full Stack Developer position at NVIDIA focusing on AI and LLM development, offering competitive salary and hybrid work arrangement.

Senior Developer Technology Engineer - AI

Senior AI Developer Technology role at NVIDIA focusing on GPU acceleration of AI workloads, requiring 8+ years experience and strong parallel programming expertise.

Senior Software Engineer, TensorRT-LLM

Senior Software Engineer position at NVIDIA working on TensorRT-LLM team, focusing on developing and optimizing machine learning inference software.

Deep Learning Performance Architect

Senior Deep Learning Performance Architect position at NVIDIA, focusing on AI workload optimization and hardware architecture analysis, requiring 2+ years of experience in parallel computing.