Senior Software Engineer for Robotics Research - Tooling

NVIDIA•Santa Clara, CA

About The Position

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.

Requirements

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

Nice To Haves

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
experience with PyTorch, Ray, Kubernetes

Responsibilities

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
Collaborate with researchers to gather requirements, understand tooling / visualization / automation needs, and deliver full-stack solutions that move the needle with speed of light.