NVIDIA Corporation

Compensation

Full Description

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 • 8+ 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! Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5. You will also be eligible for equity and benefits. 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.