Principal Architect, AI Networking Job at NVIDIA, Wausau, WI

  • NVIDIA
  • Wausau, WI

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. 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. 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. 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. #J-18808-Ljbffr

Job Tags

Similar Jobs

GrabJobs

Remote Underwriter Job at GrabJobs

 ...holistic, category-defining cyber risk management model. As a Junior Underwriter, you won't just be checking boxes; you will take complete...  ...exactly when to escalate complex risks. ~ This role is remote but requires you to be based in or willing to relocate to the... 

Evergreen Center for Health & Rehabilitation

LPN - Licensed Practical Nurse Job at Evergreen Center for Health & Rehabilitation

 ..., we invite you to join our team as a Licensed Practical Nurse (LPN)! -: What We Offer As an affiliate of National Health Care...  ...of short-term rehabilitation, skilled nursing, and post-hospital care including several named Best Of by US News & World Report... 

Mid-America LGBT Chamber of Commerce

Remote Philanthropy Prospect Research Analyst Job at Mid-America LGBT Chamber of Commerce

 ...Unity seeks a Philanthropy Research & Technology Analyst to design a strategic, tech-enabled prospect research program for identifying top donor prospects and growing annual, capital, and planned gifts. You will conduct research across individuals, corporations, and... 

Core-Mark

Night Shift Warehouse Order Selector - Earn Up to $37/hr Job at Core-Mark

 ...Performance Food Group is seeking a Warehouse Order Selector on the night shift in Glen Allen, VA. The role involves selecting cases and containers from multi-level racks, loading and staging orders for shipment, using pallets, pallet jacks, and forklifts, while adhering... 

Booz Allen Hamilton

Data Scientist Job at Booz Allen Hamilton

Data ScientistThe Opportunity: As a data scientist, youre excited at the prospect of unlocking the secrets held by a data set, and youre fascinated by...  ...performance and superior demonstration of our values. Full-time and part-time employees working at least 20 hours a week on...