Senior Research Engineer, Foundation Model Training Infrastructure

4 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

SEGA - Senior Community Manager

SEGA

England, United Kingdom (Hybrid)
2 Months ago
Sword Health - Computer Vision Trainee

Sword Health

Portugal (Hybrid)
1 Week ago
NVIDIA - Manager, Software Engineering

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
4 Months ago
NCR Voyix - Java Full Stack Developer

NCR Voyix

Hyderabad, Telangana, India (On-Site)
1 Month ago
Nordson Corporation - Technical Customer Engineer

Nordson Corporation

Bangkok, Thailand (On-Site)
1 Month ago
techholding - AWS DevOps Engineer

techholding

Mexico (Remote)
2 Months ago
Google - Senior Software Developer, Site Reliability Engineering, Google Cloud

Google

Durham, North Carolina, United States (On-Site)
7 Months ago
PwC - ETIC, Cloud Solution Architect (Multi-Cloud, DevOps Focus) - Senior Manager

PwC

Cairo, Cairo Governorate, Egypt (On-Site)
8 Months ago
Scopely - Senior Server Engineer (Platform)

Scopely

Lisbon, Lisbon, Portugal (Hybrid)
4 Months ago
Enphase Energy - Sr. Software Engineer - Enlighten Cloud Backend

Enphase Energy

Bengaluru, Karnataka, India (On-Site)
6 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Capgemini - Sitecore Developer

Capgemini

Mumbai, Maharashtra, India (On-Site)
1 Month ago
Barracuda - Technical Support Representative

Barracuda

Alpharetta, Georgia, United States (On-Site)
1 Week ago
Google - Software Engineer III, Site Reliability Engineering

Google

Warsaw, Masovian Voivodeship, Poland (On-Site)
2 Months ago
Forescout Technologies  Inc  - Senior Pre-Sales Engineer

Forescout Technologies Inc

London, England, United Kingdom (On-Site)
3 Weeks ago
PlayStation Global - Staff Linux Network Software Engineer

PlayStation Global

Berlin, Berlin, Germany (On-Site)
5 Months ago
Cadence - Lead Product Engineer

Cadence

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Interactive Brokers - Financial Services Representative

Interactive Brokers

Chicago, Illinois, United States (On-Site)
1 Month ago
ansira - Client Success Delivery Senior Associate

ansira

Noida, Uttar Pradesh, India (On-Site)
2 Days ago
Corsair gaming - Mechanical Engineer

Corsair gaming

Taipei City, Taiwan (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Jobs in Canada

ShyftLabs - Senior Backend Developer

ShyftLabs

Toronto, Ontario, Canada (Hybrid)
4 Months ago
Granicus - Critical - Technical Implementation Consultant

Granicus

Canada (Remote)
1 Month ago
Spaulding Ridge - Finance Transformation Lead

Spaulding Ridge

Toronto, Ontario, Canada (On-Site)
2 Months ago
Track VFX - Vancouver | Technical Artist

Track VFX

Vancouver, British Columbia, Canada (On-Site)
9 Months ago
Larian Studios - Sound Designer

Larian Studios

Quebec, Canada (On-Site)
3 Months ago
Thumbtack - Senior Experience Researcher

Thumbtack

Ontario, Canada (Remote)
1 Month ago
Ubisoft - Technical Art Director [Far Cry Project]

Ubisoft

Montreal, Quebec, Canada (Hybrid)
2 Months ago
Airlab Inc  - Game Designer (Mobile)

Airlab Inc

Montreal, Quebec, Canada (On-Site)
11 Months ago
Amber - Localization Quality Assurance (Korean)

Amber

Quebec, Canada (Hybrid)
3 Months ago
Scanline VFX - Senior Pipeline Developer (Houdini)

Scanline VFX

Montreal, Quebec, Canada (Remote)
8 Months ago

Get notifed when new similar jobs are uploaded

Devops Jobs

Adobe - Senior Software Engineer, Site Reliability Engineering

Adobe

Bucharest, Bucharest, Romania (On-Site)
1 Year ago
Ziff Davis - Systems/DevOps Engineer

Ziff Davis

Canada (Remote)
1 Month ago
PwC - Experienced Associate - Infrastructure Engineer

PwC

Kuala Lumpur, Federal Territory Of Kuala Lumpur, Malaysia (On-Site)
1 Month ago
FICO - Sr. Cloud DevOps Engineer (Crossplane, ArgoCD)

FICO

Bengaluru, Karnataka, India (On-Site)
3 Weeks ago
Adyen - Solutions Engineer

Adyen

Warsaw, Masovian Voivodeship, Poland (On-Site)
1 Month ago
Litmus - Partner Sales Engineer – Cloud & Industrial Analytics

Litmus

United States (Remote)
1 Week ago
Unity - Senior DevOps Developer

Unity

Montreal, Quebec, Canada (On-Site)
1 Month ago
Synechron - Scrum Master (DevOps Expertise in Cloud Computing and AI/ML Technologies)

Synechron

Pune, Maharashtra, India (On-Site)
2 Weeks ago
Scopely - Platform Account Engineer

Scopely

Mexico City, Mexico City, Mexico (Hybrid)
3 Months ago
NBC Universal - Manager, Site Reliability Engineer (ServiceNow)

NBC Universal

Englewood Cliffs, New Jersey, United States (Remote)
2 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)

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)

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