← All roles

System Software Architect, AI and GPU Networking

NVIDIA· Architect
Apply on company site
Beijing, China, Shanghai, Chinafull-timePosted 7mo ago

About the role

Senior technical architect role focused on accelerate networking and AI data centers, designing and prototyping features to optimize AI workloads, data movement, and system-level performance across AI networking and related technologies.

Responsibilities

  • Enhance NVIDIA's GPU Networking offerings for accelerating AI workloads, such as NVIDIA Dynamo, NVIDIA NIXL and NVIDIA UCX, tailored to the unique requirements of AI workloads.
  • Design and prototype features and optimizations that accelerate data movement and enable new capabilities for inference and model serving - focusing on throughput, latency, and memory efficiency.
  • Identify and evaluate new technologies, innovations and partner relationships for alignment with our technology roadmap and business value.
  • Develop and evaluate innovative features with respect to runtime systems, communication libraries, AI-specific technologies.
  • Develop and evaluate enhancements to communication libraries such as NIXL, UCX, and GPUnetIO, tailored to the unique demands of AI workloads.

Qualifications

  • Hold a M.Sc. or Ph.D. in Computer Science, Electrical or Computer Engineering from a leading university (or equivalent experience).
  • 5+ years of industry experience (or equivalent) in system architecture, AI systems architecture, scaling of AI, Parallelism of AI frameworks, or deep learning training workloads.
  • Experienced in algorithm design, system programming, computer architecture and operating systems.
  • Experienced in virtualization, networking and storage.
  • Deep understanding of performance profiling and optimization techniques, together with defining and using hardware features.
  • Strong programming background in C++, Python, and ideally CUDA or other GPU programming models.
  • Ability and flexibility to work and communicate effectively in a multi-national, multi-time-zone corporate environment.

Skills

Required:M.Sc. or Ph.D. in Computer Science, Electrical or Computer EngineeringSystem architectureAI systems architectureC++, Python
Nice to have:CUDA or GPU programming models
Last updated Jul 3, 2026. Interested? Apply on the company site.

About NVIDIA

World Leader in AI Computing

NVIDIA invents the GPU and drives advances in AI, HPC, gaming, creative design, autonomous vehicles, and robotics.

AIautonomous vehiclesgaminggpugraphics cardhigh performance computingliquid-ai-recruiting-targetllm
Company profileWebsite

More open roles at NVIDIA