Senior AI Systems Architect — High-Performance Networking

NVIDIA Corporation; Austin (TX); On-site; USD 224,000 - 431,000; Full time

Job summary

NVIDIA Corporation in Austin is seeking a Senior Architect to own end-to-end modules in high-performance distributed AI systems, spanning low-level networking, memory management, and production-ready code. You will collaborate across GPU, DPU, NIC, and switch teams, and prototype new approaches to push the boundaries between research and production.

The role emphasizes leadership in architecture, debugging, and open-source contributions, with base salary ranges of 224,000–431,250 USD plus equity

Qualifications

• 12+ years in systems software and/or networking with ownership of complex projects.
• MS, PhD or equivalent in Computer Science, Computer Engineering, Electrical Engineering, or a related field.
• Solid understanding of high-performance networking: InfiniBand, RoCE, RDMA, NVLink, GPUDirect.
• Strong C/C++/Rust systems programming with performance profiling and low-level debugging.
• Understanding of ML systems concepts—transformer architectures, KV cache mechanics, model parallelism, or distributed training and inference patterns.
• Knowledge of ML inference frameworks (vLLM, SGLang, TensorRT-LLM) and their communication requirements.
• Knowledge of storage networking (NVMe-oF, GPUDirect Storage, S3).
• Background of Reinforcement Learning systems.

Responsibilities

• Architecting and implementing high-performance communication and memory management libraries for distributed AI.
• Driving hardware-software co-optimization with GPU, DPU, NIC, and switch teams through GPUDirect RDMA, NVLink, and next-generation interconnects.
• Profiling and optimizing data movement across GPU memory, system DRAM, NVMe, and network fabrics.
• Integrating networking capabilities into AI serving stacks such as vLLM, SGLang, and TensorRT-LLM.
• Contributing to and maintaining open-source projects, mentoring engineers, conducting design reviews, and prototyping experimental technologies to evaluate their viability.

Skills

C/C++/Rust systems programming

Performance profiling

Low-level debugging

Distributed systems

ML systems concepts

Education

MS or PhD in CS/CE/EE

Job description

NVIDIA Corporation in Austin is seeking a Senior Architect to own end-to-end modules in high-performance distributed AI systems, spanning low-level networking, memory management, and production-ready code. You will collaborate across GPU, DPU, NIC, and switch teams, and prototype new approaches to push the boundaries between research and production.

The role emphasizes leadership in architecture, debugging, and open-source contributions, with base salary ranges of 224,000–431,250 USD plus equity