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 building foundational data infrastructure to support robotics research and large-scale multimodal foundation models. You will join a new team working closely with NVIDIA’s Generalist Embodied Agent Research (GEAR) group and focus on Project GR00T — an initiative to build foundation models and full-stack technology for humanoid robots. You will collaborate with researchers on multimodal foundation models, large-scale robot learning, embodied AI, and physics simulation.

Responsibilities

  • Design and maintain large-scale distributed data ETL and data management systems to support multimodal foundation models for robotics.
  • Optimize GPU and cluster utilization for efficient model training and fine-tuning on very large datasets.
  • Implement scalable data loaders and preprocessors tailored for multimodal datasets (video, text, sensor data).
  • Develop robust observability tools and procedures for the 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 such as SQL, Apache Spark, and 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 technical lead experience coordinating engineering teams and driving projects from conception to deployment.
  • Strong experience building and operating large-scale data infrastructure.
  • Strong experience and curiosity in frontier AI research and foundation models.

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.

Additional information

  • Applications will be accepted at least until October 3, 2025.
  • NVIDIA is an equal opportunity employer and is committed to fostering a diverse work environment.