Senior Software Engineer, Compute Infrastructure for Robotics Research

We are seeking a Senior Software Engineer to join a new team building the foundational infrastructure for Robotics Research. This new team will work very closely with NVIDIA’s Generalist Embodied Agent Research (GEAR) group. The near term focus is Project GR00T, NVIDIA’s moonshot initiative at building foundation models and full-stack technology for humanoid robots. This particular position will focus on compute infrastructure.

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. Your contributions will have a significant impact on our research projects and product roadmaps.

What you’ll be doing:

• Develop mechanisms to launch and manage large compute jobs to support multi-modal foundation models for robotics. These will include data jobs, training jobs, evaluation jobs, and so forth.

• Optimize GPU and cluster utilization for efficient model training, fine-tuning, and evaluation on massive datasets.

• Develop robust observability tools and procedures for this compute infrastructure to ensure reliability and performance.

• Collaborate with researchers to integrate innovative compute technologies into scalable training and eval pipelines.

What we need to see:

• Bachelor's degree in Computer Science, Engineering, or a related field, or equivalent experience

• 12+ years of full-time industry experience in large-scale MLOps and AI infrastructure

• Experience with ML frameworks like PyTorch, JAX, or TensorFlow.

• Deep understanding of Kubernetes, experience with Ray

• Experience with data frameworks and standards like SQL, Apache Spark, LanceDB

• Experience of GPU acceleration and CUDA programming

• Strong programming skills in Python and a high-performance language such as C++ for efficient system development.

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

• Deep background at building and operating large-scale data infrastructure

• Strong experience and curiosity in frontier AI research

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!

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 224,000 USD - 356,500 USD for Level 5, and 272,000 USD - 425,500 USD for Level 6.Applications for this job will be accepted at least until October 5, 2025.

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.