Senior Software Engineer for Robotics Research - Tooling

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 ML productivity tooling – we are looking for people who have high agency and love creating tools to make researchers more productive and GPUs more efficiently used.

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:

• Build highly scalable, robust, and efficient CI/CD frameworks. The workload is data intensive and requires CPU/GPU heterogeneous computation.

• Build world-class visualization tools for analyzing and optimizing for all our datasets and compute jobs (across 10s of thousands of GPUs)

• Develop and apply AI agents to significantly improve programming efficiency within the team, and decrease the human effort in fixing job failures

• Overall, collaborate with researchers to gather requirements, understand tooling / visualization / automation needs, and deliver full-stack solutions that move the needle with speed of light.

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. 

• Strong experience in full-stack software development, with a focus on building CI/CD or visualization tools.

• Proficient in both front-end and back-end programming with Python, JavaScript, SQL, or similar.

• Familiar with modern web front/back end technologies like React, Node.js.

• Knowledge of GPU technologies like CUDA and NCCL

• Bonus: experience with PyTorch, Ray, 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 and operating large-scale tooling infrastructure in production

• Strong background 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!