Outscal Logooutscal logo

Senior Research Engineer, Foundation Model Training Infrastructure

1 Month 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 multimodal 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 frameworks like PyTorch, JAX, or TensorFlow, and strong programming skills in Python and C++. The position contributes directly to Project GR00T, a moonshot initiative focused on building foundation models for humanoid robots.
Must have:
  • 10+ years MLOps/AI infrastructure experience
  • Expertise in PyTorch/JAX/TensorFlow
  • GPU acceleration, CUDA, Kubernetes knowledge
  • Python & C++ programming skills
  • Large-scale GPU cluster experience
Good to have:
  • Master's/PhD in relevant field
  • Tech Lead experience
  • LLM/Multimodal LLM infrastructure experience
  • Open-source contributions or top-tier 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

VGW - Machine Learning Engineer

VGW

Sydney, New South Wales, Australia (On-Site)
3 Weeks ago
Welevel - Senior AI Engineer (LLM & Training Focus)

Welevel

Munich, Bavaria, Germany (On-Site)
3 Weeks ago
ByteDance - Algorithm Engineer - Audio Understanding - Start 2025

ByteDance

Singapore (On-Site)
4 Months ago
SmileGate - Game Data Engineer [LOST ARK]

SmileGate

Seongnam-si, Gyeonggi-do, South Korea (On-Site)
2 Months ago
Altagram Group - Data Science Internship/Workstudent

Altagram Group

Germany (On-Site)
8 Hours ago
Google - Senior Software Engineer, Machine Learning, Google Ads

Google

Los Angeles, California, United States (On-Site)
3 Months ago
Meta - Research Scientist, Machine Learning (PhD)

Meta

Sunnyvale, California, United States (On-Site)
4 Months ago
ByteDance - Researcher - Large Language Models, Applied Machine Learning

ByteDance

Seattle, Washington, United States (On-Site)
1 Day ago
Kokotree - Artificial Intelligence Developers

Kokotree

Wilmington, North Carolina, United States (On-Site)
4 Months ago
ByteDance - Student Researcher (Doubao (Seed) - Foundation Model - MultiModal Generative Model)

ByteDance

San Jose, California, United States (On-Site)
1 Day ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Pluto7 - Data Scientist

Pluto7

Bengaluru, Karnataka, India (On-Site)
7 Months ago
ByteDance - Tech Lead - Applied Machine Learning Algorithm

ByteDance

San Jose, California, United States (On-Site)
4 Months ago
Netomi - Data Scientist - I

Netomi

Gurugram, Haryana, India (Hybrid)
5 Months ago
CharacterAI - Software Engineer, Machine Learning Infrastructure

CharacterAI

New York, New York, United States (On-Site)
12 Hours ago
Seedify - AI Product Manager

Seedify

(Remote)
2 Months ago
Rackspace Technology - Machine Learning Architect (AWS)

Rackspace Technology

(Remote)
2 Months ago
ByteDance - AI Security Researcher - Security - San Jose

ByteDance

San Jose, California, United States (On-Site)
4 Months ago
bosh group india - Asst Manager / Sr. Engineer - Data Scientist Prognostics

bosh group india

Karnataka, India (On-Site)
3 Months ago
Ubisoft - ML OPS Senior _ Groupe Technologique Création de contenu

Ubisoft

Montreal, Quebec, Canada (On-Site)
2 Months ago
Armada - Senior Data Engineer

Armada

Thiruvananthapuram, Kerala, India (On-Site)
5 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Canada

Epic Games - Senior Producer

Epic Games

Vancouver, British Columbia, Canada (On-Site)
2 Months ago
Scientific Games  - Electrotechnician

Scientific Games

Montreal, Quebec, Canada (On-Site)
3 Months ago
Epic Games - QA Lead

Epic Games

Montreal, Quebec, Canada (On-Site)
1 Day ago
Spyke Games - General Opportunities (Don't see an open role advertised that matches your skill set?)

Spyke Games

Edmonton, Alberta, Canada (Hybrid)
3 Months ago
NVIDIA - Senior Memory Controller Verification Engineer

NVIDIA

Toronto, Ontario, Canada (On-Site)
1 Month ago
Behaviour Interactive - Senior Audio Programmer - Dead by Daylight

Behaviour Interactive

Montreal, Quebec, Canada (Hybrid)
2 Months ago
Fortis Games - Lead Marketing Analyst

Fortis Games

Canada (On-Site)
4 Months ago
Epic Games - Character Art Director

Epic Games

Montreal, Quebec, Canada (On-Site)
22 Hours ago
Epic Games - Concepteur de systèmes de narration

Epic Games

Montreal, Quebec, Canada (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Zazz - Machine Learning Engineer

Zazz

(Remote)
1 Month ago
Meta - Research Intern, Computer Vision for Egocentric Representation Learning (PhD)

Meta

Redmond, Washington, United States (On-Site)
4 Months ago
ByteDance - Student Researcher (Doubao (Seed) - Foundation Model - Vision Generative AI)

ByteDance

San Jose, California, United States (On-Site)
1 Day ago
Match Group - Sr. Software Engineer, Generative AI

Match Group

Palo Alto, California, United States (Hybrid)
5 Months ago
NVIDIA - Senior Research Engineer, Foundation Model Training Infrastructure

NVIDIA

Santa Clara, California, United States (On-Site)
2 Months ago
Axon - Senior Technical Program Manager, AI

Axon

Scottsdale, Arizona, United States (Remote)
1 Month ago
ByteDance - AI Model Optimization Engineer

ByteDance

San Jose, California, United States (On-Site)
1 Day ago
Inworld AI - Forward Deployed Engineer (AI Gameplay Engineer)

Inworld AI

Mountain View, California, United States (On-Site)
5 Days ago
Talentica Software - Data Scientist

Talentica Software

India (Remote)
5 Months ago
Egnyte - Principal Machine Learning Engineer - AI

Egnyte

India (Remote)
6 Days 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