Principal Architect, AI Networking at NVIDIA | Austin, TX, US | Rezi

Principal Architect, AI Networking at NVIDIA

Principal Architect, AI Networking

NVIDIA · Austin, TX, US

1 weeks ago

Principal Architect, AI Networking

NVIDIA · Austin, TX, US

7 days ago
Resume preview

Impress employers and recruiters.
Choose from hundreds of resume examples.

Target Resume Now
Resume preview

Tailor your resume to this Principal Architect, AI Networking role.

Rezi rewrites your resume against NVIDIA's job description. Free.

Resume score gauge reading 58 out of 100

Don't guess if your resume is good enough.

See how it scores against the Principal Architect, AI Networking posting at NVIDIA — free, in seconds.

About the Role

This Principal Architect role leads the research agenda and architectural direction for how NVIDIA’s AI systems communicate at scale—across GPUs, DPUs, NICs, and heterogeneous storage. It requires someone who defines project scope from scratch, publishes original work, and translates research breakthroughs into production-grade software that ships industry-wide!

Responsibilities

  • Setting the long-term technical vision for distributed AI communication systems—GPU-to-GPU, GPU-to-storage, and cross-node data movement.
  • Conducting original research and prototyping next-generation networking solutions over RDMA, NVLink, and GPUDirect.
  • Driving hardware-software co-optimization with GPU, DPU, NIC, and network switch.
  • Investigating fundamental bottlenecks in communication runtimes for large-scale AI workloads (KV cache transfer, disaggregated prefill/decode, model parallelism).
  • Integrating networking capabilities into AI serving stacks such as vLLM, SGLang, and TensorRT-LLM.
  • Publishing findings, representing NVIDIA in industry forums and standards bodies, and mentoring senior engineers across the organization.

Requirements

  • 15+ years in systems software and/or networking with deep expertise in high-performance networking (InfiniBand, RoCE, RDMA, NVLink), communication libraries (e.g. NIXL, NCCL, UCX, MPI, NVSHMEM), and GPU accelerated systems, with track record of defining and delivering complex, cross-team technical initiatives from research concept to production.
  • MS, PhD or equivalent experience in Computer Science, Computer Engineering, Electrical Engineering, or a related field.
  • Deep understanding of computer architecture, memory hierarchies, DMA engines, and OS-level networking.
  • Understanding of ML systems concepts—transformer architectures, KV cache mechanics, model parallelism, or distributed training and inference patterns.
  • Proficiency in programming languages such as C, C++, Rust and Python.

Skills

  • High-performance networking (InfiniBand, RoCE, RDMA, NVLink)
  • Communication libraries (e.g. NIXL, NCCL, UCX, MPI, NVSHMEM)
  • GPU accelerated systems
  • Computer architecture
  • Memory hierarchies
  • DMA engines
  • OS-level networking
  • ML systems concepts
  • Transformer architectures
  • KV cache mechanics
  • Model parallelism
  • Distributed training and inference patterns
  • C
  • C++
  • Rust
  • Python
  • ML inference frameworks (vLLM, SGLang, TensorRT-LLM)
  • CUDA programming
  • NVIDIA GPU architecture

Experience Level

  • 15+ years

Education Level

  • MS
  • PhD
  • equivalent experience

Salary/Compensations

  • 272,000 USD – 431,250 USD

Benefits

  • Equity
  • Benefits package

About the Company

  • An applied research team within NVIDIA’s Networking Systems & Software Architecture group is solving some of AI’s hardest infrastructure problems.
  • The team builds systems-level software that moves data between GPUs, nodes, and storage at the speed modern AI demands—spanning low-level transport optimization, hardware-software co-design, and communication frameworks that plug directly into production AI stacks.
  • The team's charter expands into emerging domains including quantum computing interconnects.
  • NVIDIA is widely regarded as one of the most desirable technology employers in the world.
  • Our teams are composed of some of the most forward‑thinking and driven engineers in the industry, and we continue to grow rapidly.

Equal Opportunity

  • NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer.
  • As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) 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.