Principal Architect, AI Networking

NVIDIA Corporation; Santa Clara (CA); On-site; USD 272,000 - 431,000; Full time

Benefits offered by this job

Equity

Benefits package

Job summary

NVIDIA Corporation seeks a Principal Architect to lead the research agenda for how AI systems communicate at scale. You will define scope from scratch, publish original work, and translate breakthroughs into production-grade software across GPUs, DPUs, NICs, and storage.

You will set long-term vision, conduct original research on RDMA/NVLink, and drive hardware-software co-optimization with GPU/DPU/NIC ecosystems. Equity and benefits accompany a competitive salary.

Qualifications

• 15+ years in systems software and/or networking with deep expertise in high-performance networking and GPU-accelerated systems.
• PhD or MS with equivalent research track record in CS/CE/EE.
• Strong understanding of computer architecture, memory hierarchies, DMA engines, and OS networking.
• Proficiency in C, C++, Rust and Python; ability to translate research into production-grade software.

Responsibilities

• Set long-term technical vision for distributed AI communication systems across GPUs, storage, and nodes.
• Conduct original research and prototyping of next-generation networking solutions (RDMA, NVLink, GPUDirect).
• Drive hardware-software co-optimization with GPU, DPU, NIC, and switches.
• Investigate bottlenecks in communication runtimes for large-scale AI workloads.
• Integrate networking capabilities into AI serving stacks (vLLM, SGLang, TensorRT-LLM).
• Publish findings, represent NVIDIA in industry forums, and mentor senior engineers.

Skills

C

C++

Rust

Python

GPU-accelerated systems

Education

MS or PhD in CS/CE/EE

Tools

NIXL

NCCL

UCX

MPI

NVSHMEM

Job description

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. 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!

What you will be doing:

• 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.

What we need to see:

• 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.

Ways to stand out from the crowd:

• Knowledge of ML inference frameworks (vLLM, SGLang, TensorRT-LLM) and their communication requirements.
• CUDA programming and NVIDIA GPU architecture expertise.
• Proved experience influencing product strategy and technical roadmap at a senior level.
• Major open-source contributions.

With competitive salaries and a comprehensive benefits package, 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. If you are a senior data engineer passionate about building large‑scale, high‑impact data platforms, we’d love to hear from you.

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 272,000 USD - 431,250 USD. You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until April 27, 2026.

This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.

NVIDIA is committed to fostering a diverse 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.

NVIDIA pioneered accelerated computing. Today, our AI infrastructure powers global intelligence, transforming every industry. Learn more about NVIDIA.