Senior HPC & AI Networking Architect for Scalable AI

NVIDIA; Schweiz; Vor Ort; EUR 53.118 - 117.091; Vollzeit

Zusammenfassung

NVIDIA is seeking a Senior HPC and AI Network Software Architect to help build the next generation of scalable AI infrastructure, focusing on distributed training, real-time inference, and memory-efficient communication across large clusters.

You will develop software and hardware approaches, shape platform evolution, and collaborate with researchers to design systems powering the fastest AI workloads globally.

Qualifikationen

• Ph.D., or equivalent industry experience, in computer science, computer engineering, or a closely related field.
• 5+ years of experience in systems programming, parallel or distributed computing, high-performance networking, or large-scale data movement.
• Strong programming background in C++, Python, and ideally CUDA or other GPU programming models.
• Extensive hands-on experience with AI frameworks (e.g., PyTorch, TensorFlow, JAX).
• Demonstrated success in developing high-throughput, low-latency systems.

Aufgaben

• Build and evolve the architecture of scalable software systems for distributed AI training and inference, focusing on throughput, latency, resiliency, and memory efficiency across cluster-scale deployments.
• Develop and evaluate next-generation communication and runtime capabilities in libraries such as NCCL, UCX, and UCC.
• Partner with AI framework teams and internal platform teams to build integrations, explore new approaches, and improve end-to-end performance and reliability.
• Collaborate on hardware and system-level features across GPUs, DPUs, and interconnects to speed up data movement and enable new capabilities for training, inference, and model serving at scale.
• Drive innovation across runtime systems, communication libraries, and AI-specific protocol layers, turning ideas into robust implementations.

Kenntnisse

C++

Python

CUDA

PyTorch

TensorFlow

JAX

Ausbildung

PhD or equivalent

Jobbeschreibung

NVIDIA is seeking a Senior HPC and AI Network Software Architect to help build the next generation of scalable AI infrastructure, focusing on distributed training, real-time inference, and memory-efficient communication across large clusters.

You will develop software and hardware approaches, shape platform evolution, and collaborate with researchers to design systems powering the fastest AI workloads globally.