Senior Software Engineer, Data Infrastructure for Robotics Research

at Nvidia
📍 Santa Clara, United States
USD 224,000-425,500 per year
SENIOR
✅ On-site

SCRAPED

Used Tools & Technologies

Not specified

Required Skills & Competences ?

Kubernetes @ 4 Python @ 7 SQL @ 7 Spark @ 7 ETL @ 4 MLOps @ 8 TensorFlow @ 4 PyTorch @ 4 CUDA @ 4 GPU @ 4

Details

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 data infrastructure.

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.

Responsibilities

  • 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 evaluation pipelines.

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 such as PyTorch, JAX, or TensorFlow.
  • Experience with Kubernetes and Ray.
  • Deep understanding of data frameworks and standards like 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.

Ways to stand out

  • Master's or PhD 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 data infrastructure.
  • Strong experience and curiosity in frontier AI research.

Compensation & Benefits

  • Base salary ranges by level:
    • 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.

Additional information

  • Applications for this job will be accepted at least until October 3, 2025.
  • NVIDIA is committed to fostering a diverse work environment and is an equal opportunity employer. We do not discriminate 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.