Senior Research Engineer, Foundation Model Training Infrastructure (Santa Clara)

Senior Or Principal Engineer

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

Ways To Stand Out From The Crowd

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.