Senior Software Architect, AI Systems and Networking

What You Will Be Doing

• Architecting and implementing high-performance communication and memory management libraries for distributed AI.
• Driving hardware-software co-optimization with GPU, DPU, NIC, and switch teams through GPUDirect RDMA, NVLink, and next-generation interconnects.
• Profiling and optimizing data movement across GPU memory, system DRAM, NVMe, and network fabrics.
• Integrating networking capabilities into AI serving stacks such as vLLM, SGLang, and TensorRT-LLM.
• Contributing to and maintaining open-source projects, mentoring engineers, conducting design reviews, and prototyping experimental technologies to evaluate their viability.

What We Need To See

• 12+ years in systems software and/or networking with demonstrated ownership of complex projects.
• MS, PhD or equivalent experience in Computer Science, Computer Engineering, Electrical Engineering, or a related field.
• Strong understanding of high-performance networking: InfiniBand, RoCE, RDMA, NVLink, GPUDirect.
• Strong C/C++/Rust systems programming with comfort in performance profiling and low-level debugging.
• Understanding of ML systems concepts—transformer architectures, KV cache mechanics, model parallelism, or distributed training and inference patterns.

Ways To Stand Out From The Crowd

• Knowledge of ML inference frameworks (vLLM, SGLang, TensorRT-LLM) and their communication requirements.
• Knowledge of storage networking (NVMe-oF, GPUDirect Storage, S3).
• Background of Reinforcement Learning systems.

Preferred Qualifications:

• Experience with AI frameworks like vLLM, SGLang, TensorRT-LLM.
• Experience with storage networking technologies like NVMe-oF, GPUDirect Storage, S3.
• Experience with Reinforcement Learning systems.

Benefits:

• Competitive salaries
• Comprehensive benefits package

Application Deadline:

• Accepted at least until May 23, 2026

NVIDIA is an Equal Opportunity Employer. We do not discriminate against applicants due to race, religion, gender, gender identity, sexual orientation, national origin, citizenship, age, disability, veteran status, or any other characteristic protected by law.