Senior Deep Learning Training Infrastructure Architect

NVIDIA Gruppe; California (MO); On-site; USD 224,000 - 431,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is seeking a Deep Learning Software Infrastructure Engineer to build and scale training libraries and infrastructure for autonomous driving models. You will enable training on thousands of GPUs and massive datasets, collaborating across research and platform teams to improve training speed and safety.

Base salary varies by location and experience; equity and benefits are included. Applications accepted through Sept 11, 2026. This posting is for an existing vacancy.

Qualifications

• 12+ years of professional experience building and scaling high-performance distributed systems.
• Strong knowledge of AI/ML training stacks and PyTorch in large-scale environments.
• Proficiency in Python with production-grade libraries, orchestration layers, and automation tools.

Responsibilities

• Crafting, scaling, and hardening deep learning infrastructure libraries and frameworks for training on multi-thousand GPU clusters.
• Improving efficiency across the training stack: data loaders, distributed training, scheduling, and performance monitoring.
• Building robust training pipelines and libraries to handle massive video datasets and rapid experimentation.
• Collaborating with researchers, model engineers, and internal platform teams to enhance efficiency, minimize stalls, and improve training availability.
• Owning core infrastructure components such as orchestration libraries, distributed training frameworks, and fault-resilient training systems.
• Partnering with leadership to ensure infrastructure scales with growing GPU capacity and dataset size while maintaining developer efficiency and stability.

Skills

Distributed systems

Python

Performance profiling

Datacenter networking

Kubernetes

Slurm

Automation tools

Storage systems

Cross-functional

Teamwork

Education

PhD in Computer Science or related field

MS in Computer Science or related field

BS in Computer Science or related field

Tools

PyTorch

NCCL

Lustre

RoCE

Docker

Job description

NVIDIA is seeking a Deep Learning Software Infrastructure Engineer to build and scale training libraries and infrastructure for autonomous driving models. You will enable training on thousands of GPUs and massive datasets, collaborating across research and platform teams to improve training speed and safety.

Base salary varies by location and experience; equity and benefits are included. Applications accepted through Sept 11, 2026. This posting is for an existing vacancy.