Senior Software Engineer – Data center Infrastructure for Robotics Research
Location: Santa Clara, CA or US Remote
About the Role
NVIDIA is looking for a Senior Software Engineer to join a newly formed team building the foundational infrastructure that powers cutting-edge Robotics Research. You’ll be collaborating closely with our Generalist Embodied Agent Research (GEAR) group, contributing directly to Project GR00T—NVIDIA’s bold initiative to develop foundation models and full-stack technologies for humanoid robots.
This role focuses specifically on designing and scaling data infrastructure that supports large-scale, multimodal AI and robotics systems.
What You will Be Doing
- Architect, develop, and maintain large-scale distributed data ETL systems for multimodal robotics datasets.
- Optimize GPU and cluster resource utilization for efficient model training and fine-tuning.
- Build scalable data loaders and preprocessing pipelines for videos, text, and sensor data.
- Develop observability tools to monitor and maintain system reliability and performance.
- Partner with world-class researchers to integrate data infrastructure into training and evaluation workflows.
What We are Looking For
- Bachelor’s degree in Computer Science, Robotics, Engineering, or a related field—or equivalent experience.
- 12+ years of hands-on industry experience in large-scale AI infrastructure or MLOps.
- Proficiency with ML frameworks like PyTorch, TensorFlow, or JAX.
- Strong experience with Kubernetes and Ray.
- Deep knowledge of data technologies such as SQL, Apache Spark, LanceDB.
- Familiarity with GPU acceleration and CUDA programming.
- Strong coding skills in Python and a compiled language like C++.
Bonus Points (Nice to Have)
- Master’s or Ph.D. in a relevant field.
- Proven technical leadership, including mentoring and project ownership.
- Background in building and scaling data infrastructure for frontier AI.
- Passion for robotics, AI research, and the future of embodied intelligence.