N
Not Specified Permanent

Austin, Texas · USA job

Principal Architect, AI Networking

Nvidia

Austin, Texas

Job description

Overview

In this Principal Architect role, you will lead the research agenda for NVIDIA's AI system communications at scale, spanning GPUs, DPUs, NICs, and storage. You'll define long-term technical vision, translate research breakthroughs into production-grade software, and mentor senior engineers. The role blends systems software, high-performance networking, and hardware-software co-design to push data movement bottlenecks at the edge of AI infrastructure. You'll engage with industry forums and standards bodies and influence product strategy. This is a strategic, hands-on opportunity to shape how AI systems communicate across heterogeneous components.

Compensation / Benefits
  • equity
  • comprehensive benefits package
  • competitive salary
  • opportunity to influence product roadmap
  • work with leading AI infrastructure
  • remote/hybrid options where applicable
Responsibilities
  • Set long-term technical vision for distributed AI communication systems (GPU-to-GPU, GPU-to-storage, cross-node data movement)
  • Prototype and evaluate next-generation networking solutions over RDMA, NVLink, and GPUDirect
  • Drive hardware-software co-optimization across GPU, DPU, NIC, and network switch
  • Investigate bottlenecks in large-scale AI communication runtimes (KV cache transfer, disaggregated prefill/decode, model parallelism)
  • Integrate networking capabilities into AI serving stacks (e.g., vLLM, SGLang, TensorRT-LLM)
  • Publish findings, represent NVIDIA in industry forums and standards bodies, and mentor engineers
Key requirements
  • 15+ years in systems software and/or networking with deep expertise in high-performance networking (InfiniBand, RoCE, RDMA, NVLink)
  • Proven track record of delivering cross-team technical initiatives from research concept to production
  • MS, PhD or equivalent experience in Computer Science, Computer Engineering, Electrical Engineering, or related field
  • Deep understanding of computer architecture, memory hierarchies, DMA engines, and OS-level networking
  • Knowledge of ML systems concepts (transformers, KV cache, model parallelism, distributed training/inference)
  • Proficiency in C, C++, Rust, and Python
  • Strategic thinking and influence at senior levels
  • Mentoring and leadership across teams
  • Excellent collaboration and communication
  • RDMA, NVLink, GPUDirect
  • InfiniBand, RoCE, NVLink
  • NIXL, NCCL, UCX, MPI, NVSHMEM

Explore related USA jobs

Similar jobs you may like

Related roles with a similar title and location.

Related position Same country

Social Networking: Getting Paid to Party for real

IVIX.Space

Jeannette, Pennsylvania, United States · 15644

Related position Same country

Experienced MSP - Tier 2 Tech - Microsoft 365, Networking, Server

Intelligent Network & Communication Systems

Grand Blanc, Michigan, United States · 48439

Same category Same location Same country

Low Voltage Foreman

Faith Technologies

El Paso, Texas, United States · 88568

Same category Same location Same country

PMP Project Manager

BIC Recruiting

Houston, Texas, United States · 77299

Same category Same location Same country

Welder - MIG/TIG

Texas Labor, Corp

Grand Prairie, Texas, United States · 75054

Same category Same location Same country

Solar Installer

Texas Labor, Corp

Richardson, Texas, United States · 75085

Same category Same location Same country

Welder MIG

Texas Labor, Corp

Sunnyvale, Texas, United States · 75182

Same category Same location Same country

Fabrication Shop Foreman

BIC Recruiting

Baytown, Texas, United States · 77522

Need help finding a job?

Chat with our AI assistant.