Senior Research Engineer, Foundation Model Training Infrastructure

NVIDIA; Santa Clara (CA); On-site; USD 224,000 - 357,000; Full time

Job summary

NVIDIA is seeking a Senior Research Engineer to join the GEAR group, focusing on building infrastructure for foundation model training. The role involves designing large-scale distributed systems and optimizing GPU utilization, offering a chance to impact cutting-edge research in humanoid robotics and AI.

Benefits

Equity opportunities

Comprehensive benefits package

Diverse work environment

Qualifications

• 10+ years in MLOps and AI infrastructure.
• Experience with distributed training systems.
• Strong background in GPU and cluster management.

Responsibilities

• Design and maintain distributed training systems.
• Optimize GPU utilization for model training.
• Develop monitoring tools for training workflows.

Skills

MLOps

Distributed Systems

GPU Acceleration

Python

C++

Education

Bachelor's degree in Computer Science, Robotics or Engineering

Master’s or PhD in related field

Tools

CUDA

Kubernetes

PyTorch

JAX

TensorFlow

Job description

Senior Research Engineer, Foundation Model Training Infrastructure

Join to apply for the Senior Research Engineer, Foundation Model Training Infrastructure role at NVIDIA

Senior Research Engineer, Foundation Model Training Infrastructure

Join to apply for the Senior Research Engineer, Foundation Model Training Infrastructure role at NVIDIA

Direct message the job poster from NVIDIA

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.

JR1992361

Seniority level

• Seniority level

  Mid-Senior level

Employment type

• Employment type

  Full-time

Job function

• Industries

  Computer Hardware Manufacturing, Software Development, and Computers and Electronics Manufacturing

Referrals increase your chances of interviewing at NVIDIA by 2x

Sign in to set job alerts for “Senior Research Engineer” roles.

Senior Research And Development Engineer

Senior Software Engineer - Localization and Mapping (SLAM)

Mountain View, CA $204,000.00-$259,000.00 2 weeks ago

Mountain View, CA $204,000.00-$259,000.00 1 week ago

Sr. Software Engineer, Plant Modeling and Tools

Senior Software Development Engineer, Virtual Network

San Jose, CA $224,000.00-$410,000.00 6 days ago

Sr. Software Engineer, HIL Automation, Autonomy

San Jose, CA $160,000.00-$170,000.00 23 hours ago

Senior Software Engineer, Fabric Networking - GPU

Santa Clara, CA $148,000.00-$287,500.00 6 days ago

Embedded Sr. Software Development Engineer, Hardware Compute Group

Sunnyvale, CA $151,300.00-$261,500.00 1 week ago

Senior Staff, Software Engineer (C++, System Designs, Linux, Routing & Switching)

Sr. SDE - ML, SEAS, Stores Economics and Science (SEAS)

Senior Software Engineer, ASIC Verification Tools

Sr. Software Development Engineer, Kuiper Ground Gateway Services

Sunnyvale, CA $151,300.00-$261,500.00 20 hours ago

Sunnyvale, CA $150,000.00-$168,000.00 4 days ago

Senior Research and Development Engineer, Design

Redwood City, CA $120,000.00-$145,000.00 2 days ago

Sunnyvale, CA $117,000.00-$234,000.00 1 day ago

Sunnyvale, CA $198,000.00-$220,000.00 5 days ago

Menlo Park, CA $85.10-$251,000.00 2 weeks ago

Sunnyvale, CA $85.10-$251,000.00 2 weeks ago

Sr Software Engineer (Autonomous Digital Experience Management)

Senior Software Engineer- Medical Device

Senior Software Engineer, Model Evaluation - Simulation

Mountain View, CA $204,000.00-$259,000.00 4 days ago

We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.