Outscal Logooutscal logo

Senior Research Engineer for Reinforcement Learning

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

Job Summary

Job Description

NVIDIA's Generalist Embodied Agent Research (GEAR) group seeks a Senior Research Engineer for Reinforcement Learning to contribute to Project GR00T, a moonshot initiative building foundation models for humanoid robots. Responsibilities include developing large-scale reinforcement learning training frameworks 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 like Eureka, VIMA, and others, and requires expertise in reinforcement learning algorithms, large-scale GPU clusters, and deep learning frameworks (PyTorch, JAX, TensorFlow).
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 experience with PyTorch, JAX, or TensorFlow
  • Deep familiarity with reinforcement learning algorithms (PPO, SAC, Q-learning)
  • Strong experience with large-scale GPU clusters and job scheduling tools
Good to have:
  • Master's or PhD in CS, Robotics, or related field
  • Experience transferring policies from simulation to real robots
  • Contributions to open-source RL frameworks or publications in top AI conferences
  • Ability to mentor junior engineers and lead projects
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

Lucid Reality Labs - Machine Learning Engineer

Lucid Reality Labs

Poland (Remote)
2 Months ago
The Walt Disney Company - Lead Machine Learning Engineer, Personalization & Recommendations

The Walt Disney Company

Washington, United States (On-Site)
1 Day ago
ByteDance - Research Engineer Graduate (Machine Learning Sys-US) - 2024 Start (PhD)

ByteDance

Seattle, Washington, United States (On-Site)
4 Months ago
Virtuos - Senior Games Tool Engineer (Machine Learning Specialist)

Virtuos

Shanghai, Shanghai, China (On-Site)
1 Day ago
Netflix - Research Scientist (L6) - Identity Algorithms

Netflix

Los Gatos, California, United States (On-Site)
4 Months ago
NVIDIA - Senior Firmware Verification Engineer, PCIe

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
1 Week ago
ByteDance - Research Scientist, Infrastructure System Lab

ByteDance

San Jose, California, United States (On-Site)
20 Hours ago
NVIDIA - Senior ASIC Verification Engineer - GPU

NVIDIA

Austin, Texas, United States (Hybrid)
3 Weeks ago
NVIDIA - Senior Memory Controller Verification Engineer

NVIDIA

Toronto, Ontario, Canada (On-Site)
1 Month ago
ByteDance - Image Sensor Architect - Pico - San Jose

ByteDance

San Jose, California, United States (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Dolby Laboratories - Senior Foundational AI Researcher

Dolby Laboratories

Bengaluru, Karnataka, India (Hybrid)
6 Months ago
PwC - Senior AI Developer - Roma [DIG]

PwC

Rome, Lazio, Italy (On-Site)
5 Months ago
Meta - Software Engineer, Machine Learning

Meta

Redmond, Washington, United States (On-Site)
4 Months ago
Zazz - Machine Learning Engineer

Zazz

(Remote)
1 Month ago
ByteDance - Software Engineer in Large Model System Graduate (Machine Learning Sys-US) - 2024 Start (BS/MS)

ByteDance

San Jose, California, United States (On-Site)
4 Months ago
ByteDance - Software Engineer in ML Engineering Platform

ByteDance

San Jose, California, United States (On-Site)
4 Months ago
Tencent - Research Intern (NLP)

Tencent

Palo Alto, California, United States (On-Site)
1 Month ago
ByteDance - Research Scientist in Foundation Model, Speech & Audio Graduates - 2024 Start (PhD)

ByteDance

Seattle, Washington, United States (On-Site)
4 Months ago
NVIDIA - Solution Architect, Generative AI - Digital Human

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Jobs in Canada

Ubisoft - Senior ML Data Scientist

Ubisoft

Montreal, Quebec, Canada (On-Site)
2 Months ago
Amber - Localization Quality Assurance (Dutch)

Amber

Quebec, Canada (Hybrid)
5 Days ago
Epic Games - QA Programmer

Epic Games

Vancouver, British Columbia, Canada (On-Site)
1 Month ago
DNEG - Pipeline Developer

DNEG

Montreal, Quebec, Canada (Hybrid)
5 Days ago
NVIDIA - Food and Beverage Manager

NVIDIA

Canada (On-Site)
2 Weeks ago
Epic Games - Senior DevOps Programmer

Epic Games

Montreal, Quebec, Canada (On-Site)
1 Day ago
Keywords Studios (Player Support) - Expert Subtitle Translator/QCer: English to Canadian French

Keywords Studios (Player Support)

Québec City, Quebec, Canada (Remote)
1 Week ago
Tencent - Gameplay Programmer Intern

Tencent

Quebec, Canada (On-Site)
2 Weeks ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

Rivos - SOC Physical Design Verification Engineer - Full Time

Rivos

Hsinchu, Hsinchu City, Taiwan (Hybrid)
5 Months ago
NVIDIA - Senior Packaging Technical Engineer - Hardware

NVIDIA

United States (Remote)
2 Weeks ago
ByteDance - High-Performance Computing Research Scientist (Algorithm Acceleration)

ByteDance

San Jose, California, United States (On-Site)
1 Month ago
ByteDance - High-Performance Computing Research Scientist (Inference Optimization)

ByteDance

Seattle, Washington, United States (On-Site)
1 Month ago
NVIDIA - Chip Design Manager

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
3 Weeks ago
NVIDIA - Senior Manager, Hardware Engineering

NVIDIA

Santa Clara, California, United States (Hybrid)
2 Months ago
NVIDIA - DFX Methodology Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
1 Day ago
NVIDIA - Technical Product Manager - AV Simulation

NVIDIA

Santa Clara, California, United States (On-Site)
2 Months ago
NVIDIA - Senior Emulation Power Engineer

NVIDIA

Canada (On-Site)
1 Month ago
Tesla - Senior Concepts & Layouts Engineer

Tesla

Prüm, Rhineland-Palatinate, Germany (On-Site)
1 Month 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.


Hsinchu, Hsinchu City, Taiwan (On-Site)

Yokne'am Illit, North District, Israel (On-Site)

Seoul, South Korea (Hybrid)

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)

Ra'anana, Center District, Israel (On-Site)

Shanghai, Shanghai, China (On-Site)

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)

Be'er Sheva, South District, Israel (On-Site)

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

Santa Clara, California, United States (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