Senior Research Engineer for Reinforcement Learning

4 Months ago • 5 Years + • Research & Development • $148,000 PA - $287,500 PA

Job Summary

Job Description

NVIDIA's GEAR group seeks a Senior Research Engineer specializing in large-scale reinforcement learning for their Project GR00T, focused on building foundation models for humanoid robots. Responsibilities include developing a large-scale RL training framework for thousands of GPUs, optimizing simulation infrastructure (Isaac Lab), creating sim-to-real transfer pipelines for robot deployment, and proposing scalable solutions combining LLMs with policy learning. The role involves working with cutting-edge technologies and contributing to influential research projects like Eureka and VIMA. Experience with RL algorithms (PPO, SAC, Q-learning), hyperparameter tuning, policy learning techniques, and large-scale GPU clusters is crucial.
Must have:
  • 5+ years experience in large-scale deep learning or MLOps
  • Exceptional engineering skills in building scalable distributed GPU training frameworks
  • Proficiency in Python and deep learning frameworks (PyTorch, JAX, or TensorFlow)
  • Deep understanding of reinforcement learning algorithms and techniques
  • Strong experience with large-scale GPU clusters and job scheduling tools
Good to have:
  • Master's or PhD in relevant field
  • Experience transferring policies from simulation to real robots
  • Contributions to open-source RL frameworks or publications in top AI conferences
  • Mentoring and project leadership skills
Perks:
  • Equity
  • Benefits

Job Details

NVIDIA is searching for a senior or principal engineer who specializes in large-scale reinforcement learning and policy learning in the Generalist Embodied Agent Research (GEAR) group. Our team is leading Project GR00T, NVIDIA’s moonshot initiative at building foundation models and full-stack technology for humanoid robots. 

You will work with an amazing and collaborative research team that consistently produces influential works on multimodal foundation models, large-scale robot learning, embodied AI, and physics simulation. Our past projects include Eureka, VIMA, Voyager, MineDojo, MimicPlay, Prismer, and more. Your contributions will have a significant impact on our research projects and product roadmaps.

What you will be doing:

  •  Develop a large-scale reinforcement learning training framework capable of running on thousands of GPUs; 

  • Build and optimize simulation infrastructure (based on GPU-accelerated simulators like Isaac Lab) to support the training of locomotion and manipulation policies for robots at scale;

  • Develop sim-to-real transfer pipelines and work closely with the robotics team to deploy to physical robots;

  • Propose scalable solutions that combine LLMs with policy learning. Example work: Eureka;

  • Apply reinforcement learning to finetune multimodal LLMs. 

What we need to see:

  • Bachelor’s degree or above in Computer Science, Robotics, Engineering, or a related field;

  • 5+ years of industry experience on large-scale deep learning or MLOps;

  • Exceptional engineering skills in building, testing, and maintaining scalable distributed GPU training frameworks;

  • Proficiency in Python. Hands-on model training experience in 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;

  • Strong experience with large-scale GPU clusters, HPC environments, and job scheduling/orchestration tools (e.g., SLURM, Kubernetes).

Ways to stand out from the crowd:

  • Master’s or PhD’s degree in Computer Science, Robotics, Engineering, or a related field;

  • Demonstrated experience transferring policies from simulation to real robots for locomotion and manipulation;

  • Contributions to popular open-source reinforcement learning frameworks or research publications in top-tier AI conferences, such as NeurIPS, ICRA, ICLR, CoRL;

  • Strong ability to mentor junior engineers or researchers and lead technical projects from conception to completion.

NVIDIA is widely considered to be one of the technology world's most desirable employers. We have some of the most forward-thinking and productive people in the world. Please join us and be part of the forefront of developing general-purpose robots and large-scale foundation models! 

The base salary range is 148,000 USD - 287,500 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

You will also be eligible for equity and benefits. NVIDIA accepts applications on an ongoing basis.

NVIDIA is committed to fostering a diverse 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.

Similar Jobs

Inworld AI - AI Trainer (Contractor) - Writing & Gaming

Inworld AI

Vancouver, British Columbia, Canada (Remote)
2 Months ago
Ion - Web UI Developer (React), Budapest

Ion

Budapest, Hungary (Hybrid)
7 Months ago
MIQ Digital - Account Manager - China

MIQ Digital

Guangzhou, Guangdong Province, China (Hybrid)
1 Month ago
miniclip - IOS Engineer

miniclip

Lisbon, Lisbon, Portugal (On-Site)
1 Month ago
Doiq - Technical Artist

Doiq

United States (Remote)
5 Months ago
Corsair - Senior Embedded Hardware Engineer

Corsair

Landshut, Bavaria, Germany (On-Site)
1 Month ago
Assystems - Ingénieur électricité H/F

Assystems

Montrouge, Île-de-France, France (On-Site)
7 Months ago
Tesla - Electrical Assembly Supervisor

Tesla

Rhineland-Palatinate, Germany (On-Site)
3 Months ago
NVIDIA - STA Engineer

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
4 Months ago
NVIDIA - Physical Design Manager

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Intel  - Analog Circuit Design Engineer

Intel

Toronto, Ontario, Canada (On-Site)
2 Weeks ago
foxi ventures - Marketing Artist

foxi ventures

Adelaide, South Australia, Australia (On-Site)
2 Weeks ago
Unity - Technical Product Manager - Mobile Platforms

Unity

Brighton And Hove, England, United Kingdom (On-Site)
1 Month ago
UXBERT Labs - AR/IoT Development Specialist

UXBERT Labs

Riyadh, Riyadh Province, Saudi Arabia (Hybrid)
5 Months ago
welevel  - Technical Animator

welevel

Munich, Bavaria, Germany (On-Site)
3 Months ago
playrix  - Senior Game UI Developer

playrix

Portugal (Remote)
7 Months ago
Penrose studios - Production Artist

Penrose studios

San Francisco, California, United States (On-Site)
3 Weeks ago
Rocket Science - Software Engineer - UI

Rocket Science

Cardiff, Wales, United Kingdom (Hybrid)
2 Months ago
Unity - Client Partner, Ad Monetization

Unity

San Francisco, California, United States (On-Site)
1 Month ago
Ubisoft - Technical Art Director [Far Cry Project]

Ubisoft

Montreal, Quebec, Canada (Hybrid)
1 Month ago

Get notifed when new similar jobs are uploaded

Jobs in Santa Clara, California, United States

Keywords Studios - Clinical Trial Labeling Compliance Specialist

Keywords Studios

Indiana, United States (On-Site)
2 Months ago
Biestas - Business Developer

Biestas

New York, New York, United States (On-Site)
1 Month ago
Riot Games - Program Manager III - Quantitative Research Infrastructure

Riot Games

Los Angeles, California, United States (On-Site)
3 Weeks ago
Marvell - Global Compensation Program Manager

Marvell

Santa Clara, California, United States (On-Site)
5 Days ago
Dream world  - Internship: Game Developer (Summer-Fall)

Dream world

Redwood City, California, United States (On-Site)
2 Weeks ago
Visual Concepts - Senior Server Engineer

Visual Concepts

Austin, Texas, United States (On-Site)
2 Months ago
Telastra - Part Time Customer Service & Sales Consultant

Telastra

Esperance, New York, United States (On-Site)
1 Week ago
MarketScale - UGC Consultant

MarketScale

United States (Remote)
1 Month ago
luxsoft - Software Developer T2

luxsoft

New York, New York, United States (On-Site)
3 Days ago
lifechruh - Generosity Pastor

lifechruh

Edmond, Oklahoma, United States (On-Site)
4 Weeks ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

Meta - Software Engineer, Machine Learning

Meta

New York, New York, United States (On-Site)
6 Months ago
NVIDIA - Signal Processing Communication Engineer

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
3 Months ago
bytedance - Research Scientist Intern (Doubao (Seed) - Machine Learning System) - 2025 Summer (PhD)

bytedance

Seattle, Washington, United States (On-Site)
7 Months ago
rivos - CPU Design Verification - Full-time

rivos

Hsinchu, Hsinchu City, Taiwan (Hybrid)
7 Months ago
Passive Logic - Senior Electrical Engineer

Passive Logic

Salt Lake City, Utah, United States (On-Site)
7 Months ago
bytedance - Video Codec Firmware Engineer - Multimedia Lab

bytedance

San Jose, California, United States (On-Site)
7 Months ago
Samsung Semiconductor - Staff Engineer, Machine Learning

Samsung Semiconductor

San Jose, California, United States (Hybrid)
2 Months ago
bytedance - Software Engineer, Model Interference

bytedance

San Jose, California, United States (On-Site)
4 Months ago
NVIDIA - Senior Firmware PHY Verification Engineer

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
4 Months ago
NVIDIA - System Software Application Engineer

NVIDIA

Taipei City, Taiwan (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

About The Company

Since its founding in 1993, NVIDIA (NASDAQ: NVDA) has been a pioneer in accelerated computing. The company’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined computer graphics, ignited the era of modern AI and is fueling the creation of the metaverse. NVIDIA is now a full-stack computing company with data-center-scale offerings that are reshaping industry.

Santa Clara, California, United States (On-Site)

Massachusetts, United States (On-Site)

Santa Clara, California, United States (On-Site)

Santa Clara, California, United States (On-Site)

Santa Clara, California, United States (On-Site)

Texas, United States (On-Site)

Santa Clara, California, United States (Hybrid)

Santa Clara, California, United States (Hybrid)

Pune, Maharashtra, India (On-Site)

Taipei City, Taiwan (On-Site)

View All Jobs

Get notified when new jobs are added by NVIDIA

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug