Senior Research Engineer, Foundation Model Training Infrastructure

3 Months ago • 10 Years + • Artificial Intelligence • Research & Development • $224,000 PA - $356,500 PA

Job Summary

Job Description

NVIDIA's GEAR group seeks a Senior Research Engineer to design and maintain large-scale distributed training systems for multi-modal foundation models in robotics. This role involves optimizing GPU and cluster utilization, implementing scalable data loaders, developing monitoring tools, and collaborating with researchers. The successful candidate will have 10+ years of experience in large-scale MLOps and AI infrastructure, expertise in PyTorch/JAX/TensorFlow, CUDA programming, and cluster management tools like Kubernetes. The work contributes directly to NVIDIA's Project GR00T, a moonshot initiative in humanoid robot technology.
Must have:
  • 10+ years MLOps/AI infrastructure experience
  • Distributed training system design (PyTorch/JAX/TensorFlow)
  • GPU acceleration, CUDA, Kubernetes expertise
  • Python & C++ programming skills
  • Large-scale GPU cluster experience
Good to have:
  • Master's/PhD in CS/Robotics/Engineering
  • Tech Lead experience
  • LLM/multimodal LLM infrastructure experience
  • Open-source contributions or publications
Perks:
  • Equity
  • Benefits

Job Details

NVIDIA is searching for a senior or principal engineer who specializes in building cutting-edge infrastructure for large-scale foundation model training 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:

  • Design and maintain large-scale distributed training systems to support multi-modal foundation models for robotics.

  • Optimize GPU and cluster utilization for efficient model training and fine-tuning on massive datasets.

  • Implement scalable data loaders and preprocessors tailored for multimodal datasets, including videos, text, and sensor data.

  • Develop robust monitoring and debugging tools to ensure the reliability and performance of training workflows on large GPU clusters.

  • Collaborate with researchers to integrate cutting-edge model architectures into scalable training pipelines.

What we need to see:

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

  • 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 understanding of GPU acceleration, CUDA programming, and cluster management tools like Kubernetes.

  • Strong programming skills in Python and a high-performance language such as C++ for efficient system development.

  • 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 Tech Lead experience, coordinating a team of engineers and driving projects from conception to deployment;

  • Strong experience at building large-scale LLM and multimodal LLM training infrastructure;

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

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 224,000 USD - 356,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

Voodoo - Marketing Video Artist - Gaming

Voodoo

İstanbul, Türkiye (On-Site)
5 Months ago
Playrix - Senior VFX Artist

Playrix

Georgia (Remote)
6 Months ago
Riot Games - Manager, Software Engineering - Player Platform SDK

Riot Games

Los Angeles, California, United States (On-Site)
5 Months ago
Handy Games - Mobile Rendering Engineer

Handy Games

Giebelstadt, Bavaria, Germany (On-Site)
1 Month ago
QUANTIC DREAM - Développeur UI C++ (H/F/NB)

QUANTIC DREAM

Paris, Île-de-France, France (Hybrid)
6 Months ago
Zoox - Senior/Staff Software Engineer, ML Performance Optimization

Zoox

Foster City, California, United States (On-Site)
6 Months ago
ByteDance - LLM Software Engineer/Researcher (Applied Machine Learning)

ByteDance

Seattle, Washington, United States (On-Site)
1 Month ago
ByteDance - Backend Engineer (Model Inference), Machine Learning Systems

ByteDance

Singapore (On-Site)
6 Months ago
Inworld AI - AI Trainer (Contractor) - Writing & Gaming

Inworld AI

Vancouver, British Columbia, Canada (Remote)
1 Month ago
Google - Senior Photonics Engineer, Platforms Infrastructure

Google

Sunnyvale, California, United States (On-Site)
2 Weeks ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

CD PROJEKT RED - Scanning Lead

CD PROJEKT RED

Warsaw, Masovian Voivodeship, Poland (On-Site)
2 Months ago
Meta - Software Engineer, Machine Learning

Meta

Seattle, Washington, United States (On-Site)
5 Months ago
Virtuos - Senior Producer

Virtuos

Ireland (Hybrid)
1 Month ago
albi online  - Technical UI Designer

albi online

Berlin, Berlin, Germany (Hybrid)
1 Day ago
Kwalee - Senior Game Artist

Kwalee

Royal Leamington Spa, England, United Kingdom (On-Site)
1 Month ago
Inworld AI - Senior Software Development Engineer in Test (SDET) – Game Engine SDKs - USA

Inworld AI

Mountain View, California, United States (On-Site)
6 Months ago
Life church - Ruby Staff Engineer

Life church

Edmond, Oklahoma, United States (On-Site)
6 Months ago
Colo pl - Server-Side Engineer (Game Industry Experience Not Required)

Colo pl

Minato City, Tokyo, Japan (On-Site)
1 Year ago
Google - Wafer Fabrication Manufacturing Engineer

Google

Fremont, California, United States (On-Site)
2 Days ago
Google - Quantum Algorithms Numerics Specialist

Google

San Francisco, California, United States (On-Site)
2 Days ago

Get notifed when new similar jobs are uploaded

Jobs in Santa Clara, California, United States

Google - Senior Technical Program Manager I, Security, Google Cloud Platforms

Google

Sunnyvale, California, United States (On-Site)
2 Weeks ago
Moonbug Entertainment - Senior Designer

Moonbug Entertainment

California, United States (On-Site)
3 Weeks ago
Google - Senior Product Manager, Sales Intelligence

Google

Mountain View, California, United States (On-Site)
2 Weeks ago
The EW. Scripps Company - Editor

The EW. Scripps Company

Cleveland, Ohio, United States (On-Site)
18 Hours ago
Nintendo - Senior Software Engineer (Python/React)

Nintendo

Redmond, Washington, United States (Hybrid)
8 Months ago
Meta - Software Engineering Manager, Product

Meta

Redmond, Washington, United States (Remote)
2 Weeks ago
Clear Watery Analytics - Implementations Team Lead

Clear Watery Analytics

Boise, Idaho, United States (On-Site)
19 Hours ago
Nintendo - Manager, Communications Strategy - PDR

Nintendo

Redmond, Washington, United States (Hybrid)
9 Months ago
Google - Customer Engineer II, AI/ML, Google Cloud

Google

Atlanta, Georgia, United States (On-Site)
2 Days ago
On Location - Manager, Ticket Operations - FIFA World Cup 26™

On Location

Dallas, Texas, United States (Remote)
2 Months ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Google - Customer Solutions Engineer

Google

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
2 Weeks ago
Microsoft - Senior Researcher – Generative AI

Microsoft

Redmond, Washington, United States (On-Site)
1 Week ago
ByteDance - Research Scientist Graduate (Foundation Model - Vision and Language)

ByteDance

Seattle, Washington, United States (On-Site)
1 Month ago
Keywords Studios - Research Associate - AI

Keywords Studios

(Remote)
1 Month ago
Google - AI Sales Specialist III, ISV, West

Google

San Francisco, California, United States (On-Site)
2 Weeks ago
Passive Logic - AI Control Theory & Optimization Scientist

Passive Logic

Salt Lake City, Utah, United States (On-Site)
4 Months ago
NVIDIA - Principal Technical Program Manager, AI and Enterprise Apps

NVIDIA

Santa Clara, California, United States (On-Site)
2 Weeks ago
Scopely - Senior AI Creative (Motion) - Monopoly Go

Scopely

Barcelona, Catalonia, Spain (Hybrid)
1 Month ago
Meta - Research Scientist, Machine Learning (PhD)

Meta

Bellevue, Washington, United States (On-Site)
2 Weeks ago
Microsoft - Language Engineer

Microsoft

Mountain View, California, United States (Hybrid)
2 Weeks 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)

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 (On-Site)

Santa Clara, California, United States (Hybrid)

Santa Clara, California, United States (Hybrid)

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