Senior Research Engineer, Foundation Model Training Infrastructure

5 Months ago • 10 Years + • Devops • $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

Canonical - Cloud Professional Services Manager

Canonical

(Remote)
2 Months ago
GoMotive - Technical Support Engineer, Tracking & Telematics

GoMotive

Pakistan (Remote)
3 Months ago
Trellix - Senior Software Engineer

Trellix

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Capco - AML Event Driven Review Specialist

Capco

Kuala Lumpur, Federal Territory Of Kuala Lumpur, Malaysia (On-Site)
1 Month ago
DNEG - Rigging Developer

DNEG

Montréal, Québec, Canada (On-Site)
1 Year ago
Canva - Staff Frontend Engineer - Apps API Platform

Canva

Sydney, New South Wales, Australia (Remote)
2 Months ago
Unisys - Presales Solution Architect

Unisys

Hungary (On-Site)
2 Months ago
FICO - Sr. Cloud Engineer

FICO

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Zuora - Sr Enterprise Solution Architect-Zuora Billing & CPQ

Zuora

United States (Remote)
2 Months ago
Adobe - Software Engineer - Cloud Foundation

Adobe

Mountain View, California, United States (On-Site)
4 Weeks ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Autodesk - C++ Software Developer

Autodesk

Poland (Remote)
2 Months ago
Next Level Business Services - SAP MM Consultant

Next Level Business Services

Commerce, California, United States (On-Site)
9 Months ago
Sporty - Programming Scheduler

Sporty

Rio De Janeiro, Brazil (On-Site)
1 Month ago
Coherent corp. - Process Engineer - Machining

Coherent corp.

Monroe, Connecticut, United States (On-Site)
2 Months ago
Coupa - Sr. IT Support Specialist

Coupa

Bogota, Colombia (Hybrid)
2 Months ago
Ion - NetSuite ERP Developer

Ion

London, England, United Kingdom (On-Site)
5 Months ago
Ion - Senior Business Consultant - Endur

Ion

Houston, Texas, United States (On-Site)
9 Months ago
Palo Alto Networks - Sr Staff Engineer Customer Success

Palo Alto Networks

Amsterdam, North Holland, Netherlands (On-Site)
2 Months ago
Casumo - Junior IT System Administrator

Casumo

Skopje, Greater Skopje, North Macedonia (On-Site)
1 Month ago
Scanline VFX - Modeler

Scanline VFX

Seoul, South Korea (Hybrid)
9 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Canada

Epic Games - Lead Online Programmer

Epic Games

Montreal, Quebec, Canada (On-Site)
3 Months ago
reflector entertainment - Senior Lighting Artist

reflector entertainment

Montréal, Québec, Canada (Hybrid)
4 Weeks ago
Blazesoft - Senior Data Analyst (Gaming)

Blazesoft

Vaughan, Ontario, Canada (On-Site)
3 Months ago
Rackspace Technology - Engagement Manager

Rackspace Technology

Canada (Remote)
1 Month ago
Workato - Engagement Manager

Workato

British Columbia, Canada (On-Site)
2 Months ago
magmic - Senior Unity Game Developer

magmic

Ottawa, Ontario, Canada (Hybrid)
3 Months ago
Epic Games - Senior Software Engineer

Epic Games

Canada (On-Site)
4 Months ago
Critical mass - Marketing Science Senior Analyst

Critical mass

Calgary, Alberta, Canada (On-Site)
4 Weeks ago
WebTech Corporation - Supervisor - Welding

WebTech Corporation

Cambridge, Ontario, Canada (On-Site)
1 Month ago
CAE - Internal Auditor

CAE

Montreal, Quebec, Canada (Hybrid)
7 Months ago

Get notifed when new similar jobs are uploaded

Devops Jobs

Axon - Solutions Architect II

Axon

Scottsdale, Arizona, United States (Hybrid)
2 Months ago
Thales - IVV Automation Engineer

Thales

Vélizy-Villacoublay, Île-de-France, France (On-Site)
2 Months ago
Tencent - DevOps Engineer Intern

Tencent

(On-Site)
5 Months ago
Google - Software Engineer III, Infrastructure, Core

Google

Seattle, Washington, United States (On-Site)
3 Months ago
Apple - Accessibility Software Automation Engineer

Apple

Cupertino, California, United States (On-Site)
1 Month ago
ShyftLabs - Salesforce Marketing Cloud Architect

ShyftLabs

Noida, Uttar Pradesh, India (Hybrid)
4 Months ago
Zones - Inside Solutions Architect - UC & Collaboration

Zones

Islamabad, Islamabad Capital Territory, Pakistan (On-Site)
1 Month ago
Salesforce - Account Executive, Cloud Sales (Service Cloud) GRB Osaka

Salesforce

Osaka, Osaka, Japan (On-Site)
7 Months ago
Virtusa - DevOps Lead

Virtusa

Pune, Maharashtra, India (Hybrid)
9 Months ago
zoox - Software Engineer, Metrics Tooling and Automation

zoox

Foster City, California, United States (Hybrid)
3 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.

Taipei City, Taiwan (On-Site)

Beijing, Beijing, China (On-Site)

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

Santa Clara, California, United States (Hybrid)

Bengaluru, Karnataka, India (Hybrid)

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

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

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

Dubai, Dubai, United Arab Emirates (On-Site)

Beijing, Beijing, China (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