Senior Software Engineer, Compute Infrastructure for Robotics Research
at Nvidia
📍 Santa Clara, United States
USD 224,000-425,500 per year
SCRAPED
Used Tools & Technologies
Not specified
Required Skills & Competences ?
Kubernetes @ 7 Python @ 7 SQL @ 4 Spark @ 4 MLOps @ 8 TensorFlow @ 4 PyTorch @ 4 CUDA @ 4 GPU @ 4Details
We are seeking a Senior Software Engineer to join a new team building the foundational infrastructure for Robotics Research. This 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 to build foundation models and full-stack technology for humanoid robots. This position focuses on compute infrastructure.
You will work with a collaborative research team producing influential work on multimodal foundation models, large-scale robot learning, embodied AI, and physics simulation. Your contributions will impact research projects and product roadmaps.
Responsibilities
- Develop mechanisms to launch and manage large compute jobs to support multi-modal foundation models for robotics, including data jobs, training jobs, evaluation jobs, and more.
- Optimize GPU and cluster utilization for efficient model training, fine-tuning, and evaluation on massive datasets.
- Develop robust observability tools and procedures for the compute infrastructure to ensure reliability and performance.
- Collaborate with researchers to integrate innovative compute technologies into scalable training and evaluation pipelines.
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.
- Experience with ML frameworks such as PyTorch, JAX, or TensorFlow.
- Deep understanding of Kubernetes; experience with Ray.
- Experience with data frameworks and standards such as SQL, Apache Spark, LanceDB.
- Experience with GPU acceleration and CUDA programming.
- Strong programming skills in Python and a high-performance language such as C++ for efficient system development.
Preferred / Ways to stand out
- Master’s or PhD in Computer Science, Robotics, Engineering, or a related field.
- Demonstrated tech lead experience coordinating engineering teams and driving projects from conception to deployment.
- Deep background building and operating large-scale data infrastructure.
- Strong experience and curiosity in frontier AI research.
Compensation & Benefits
- Base salary range:
- Level 5: 224,000 USD - 356,500 USD
- Level 6: 272,000 USD - 425,500 USD
- You will also be eligible for equity and benefits (see NVIDIA benefits page: https://www.nvidia.com/en-us/benefits/).
Other details
- Applications for this job will be accepted at least until October 5, 2025.
- NVIDIA is an equal opportunity employer and is committed to fostering a diverse work environment.