Principal Architect, AI Networking

Time Type: Full time 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. 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 See15+ 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 CrowdKnowledge 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. You will also be eligible for equity and benefits. Applications for this job will be accepted at least until April 27, 2026.

Create an account to see the full posting, access our search engine, and more.