Join a new team building the foundational infrastructure for Robotics Research and work closely with NVIDIA’s Generalist Embodied Agent Research (GEAR) group.
Focus on Project GR00T: building foundation models and full-stack technology for humanoid robots, with emphasis on data infrastructure.
Design and maintain large-scale distributed data ETL and data management 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 observability tools and procedures for this data infrastructure to ensure reliability and performance.
Collaborate with researchers to integrate cutting-edge data technologies into scalable training and eval pipelines.
Contribute to research projects and product roadmaps with significant impact on NVIDIA's robotics initiatives.
Requirements
Bachelor's degree in Computer Science, Robotics, 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.
Experience with Kubernetes and Ray.
Deep understanding of 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++.
Master’s or PhD’s degree in Computer Science, Robotics, Engineering, or a related field (preferred).
Demonstrated Tech Lead experience, coordinating a team of engineers and driving projects from conception to deployment (preferred).
Strong experience at building and operating large-scale data infrastructure (preferred).
Strong experience and curiosity in frontier AI research (preferred).
Benefits
You will also be eligible for equity and benefits.
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.