Principal Architect, AI Inference Networking - NIXL and Dynamo

Posted 4 Hours Ago
Be an Early Applicant
2 Locations
In-Office
Expert/Leader
Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
The Role
Lead NIXL adoption and architecture for large-scale AI inference deployments in China. Profile and optimize distributed inference stacks, including KV-cache movement, memory registration, transfer scheduling, and GPU data paths. Develop NIXL backends and plugins, extend Dynamo integrations, define dynamic APIs for elastic worker and runtime routing changes, contribute upstream, and influence NVIDIA’s roadmap. Build regional community adoption through benchmarks, talks, and reference architectures while collaborating across China, Israel, and the US.
Summary Generated by Built In

Inference has become a networking problem. With disaggregated serving, where prefill runs on one set of GPUs, decode on another, and KV cache streams between them, performance depends on how fast you can move tensors between GPUs, across nodes, and in and out of storage. Performance is only half of it. These deployments are elastic, so the data path needs dynamic APIs that let workers join and leave, connections form and tear down, and transfers reroute at runtime without stalling the pipeline. NIXL, NVIDIA's open source Inference Xfer Library, is the layer that solves this, and it sits underneath NVIDIA Dynamo and a growing list of inference engines.


China is where some of the world's most demanding inference deployments are being built. We are hiring a senior architect to work directly with the platform teams building them, understanding what breaks at scale and turning that into upstream code and roadmap. You will work daily with core NIXL and Dynamo engineers in Israel and the US.

What you'll be doing:

  • Drive NIXL adoption in China, from the first architecture conversation through prototype and benchmark to production rollout and upstream contribution.
  • Profile customer inference stacks and fix what's costing throughput and TTFT: KV cache movement, memory registration, transfer scheduling.
  • Write NIXL backends and plugins, extend Dynamo integrations, and contribute optimizations upstream to NIXL and GPUDirect class data paths.
  • Help define the APIs that make the data path dynamic, supporting elastic scaling, worker churn and runtime rerouting in production inference clusters.
  • Bring the region's requirements into NVIDIA's roadmap with enough technical authority to change it.
  • Grow the local community around NIXL through talks, benchmarks and reference architectures.

What we need to see:

  • M.Sc. or Ph.D. in CS/EE/CE, or equivalent depth earned in industry.
  • 12+ years building or optimizing large scale distributed systems: AI inference or training infrastructure, communication libraries, high performance networking, or HPC runtimes.
  • Strong systems programming in C++ and Python, with performance critical code shipped.
  • Hands on experience with RDMA networking (InfiniBand or RoCE) and GPU memory movement.
  • Working knowledge of modern LLM serving, including prefill/decode disaggregation, KV cache management, tensor and pipeline parallelism, and at least one serving stack (Dynamo, TensorRT-LLM, vLLM, SGLang, Triton).
  • Strong communication skills, with the technical credibility to be taken seriously by customer engineering teams and by NVIDIA's core library maintainers.
  • Fluent Mandarin and professional English, and comfortable working across China, Israel and US time zones.

Ways to stand out from the crowd:

  • Open source contributions to NIXL, Dynamo, NCCL, vLLM, SGLang, TensorRT-LLM or Mooncake. We will read your commits.
  • Inference at genuine scale: thousands of GPUs, multi tenant clusters, strict SLOs.
  • CUDA, and comfort reading kernel and driver level code when the answer is down there.
  • A community you've built: a project people adopted, a talk people still cite.

You will help shape how NIXL is adopted across China, working alongside the engineers who design the transport layer itself. The work is public, and disaggregated inference is being defined right now.


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.

Skills Required

  • M.Sc. or Ph.D. in Computer Science, Electrical Engineering, Computer Engineering, or equivalent industry experience
  • 12+ years building or optimizing large-scale distributed systems, AI infrastructure, communication libraries, high-performance networking, or HPC runtimes
  • Strong systems programming experience in C++ and Python
  • Experience shipping performance-critical code
  • Hands-on experience with RDMA networking, including InfiniBand or RoCE
  • Experience with GPU memory movement
  • Working knowledge of modern LLM serving, including prefill/decode disaggregation, KV-cache management, tensor parallelism, and pipeline parallelism
  • Experience with at least one serving stack: Dynamo, TensorRT-LLM, vLLM, SGLang, or Triton
  • Strong communication skills and technical credibility with customers and library maintainers
  • Fluent Mandarin and professional English
  • Comfort working across China, Israel, and US time zones
  • Open-source contributions to NIXL, Dynamo, NCCL, vLLM, SGLang, TensorRT-LLM, or Mooncake
  • Experience operating inference at thousands-of-GPUs scale, with multi-tenant clusters and strict SLOs
  • CUDA experience and ability to read kernel- and driver-level code
  • Experience building an adopted technical community, project, or widely cited presentation

NVIDIA Compensation & Benefits Highlights

The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about NVIDIA and has not been reviewed or approved by NVIDIA.

  • Equity Value & Accessibility Equity awards and a discounted ESPP are highlighted as core parts of total compensation, enabling employees to share in the company’s success. Stock-based compensation and the two-year lookback ESPP are consistently described as especially valuable.
  • Healthcare Strength Health coverage is portrayed as robust, with comprehensive medical, dental, and vision options alongside mental health support and on-site care resources. Employer HSA contributions and wellness perks reinforce the depth of the offering.
  • Retirement Support Retirement programs are depicted as strong, featuring a meaningful 401(k) match with Roth options and support for Mega Backdoor Roth contributions. These elements position long-term savings as a notable advantage of the total rewards package.

NVIDIA Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Santa Clara, CA
21,960 Employees
Year Founded: 1993

What We Do

NVIDIA’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. More recently, GPU deep learning ignited modern AI — the next era of computing — with the GPU acting as the brain of computers, robots, and self-driving cars that can perceive and understand the world. Today, NVIDIA is increasingly known as “the AI computing company.”

Similar Jobs

UL Solutions Logo UL Solutions

Inspector CRS

Automotive • Professional Services • Software • Consulting • Energy • Chemical • Renewable Energy
Hybrid
Shenzhen, Guangdong, CHN
15000 Employees

UL Solutions Logo UL Solutions

Project Engineer

Automotive • Professional Services • Software • Consulting • Energy • Chemical • Renewable Energy
Hybrid
Guangzhou, Guangdong, CHN
15000 Employees

UL Solutions Logo UL Solutions

Inspector CRS

Automotive • Professional Services • Software • Consulting • Energy • Chemical • Renewable Energy
Hybrid
Shenzhen, Guangdong, CHN
15000 Employees

UL Solutions Logo UL Solutions

Inspector CRS

Automotive • Professional Services • Software • Consulting • Energy • Chemical • Renewable Energy
Hybrid
Shenzhen, Guangdong, CHN
15000 Employees

Similar Companies Hiring

Revel Thumbnail
Aerospace • Hardware • Robotics • Software
Marina Del Rey, California
60 Employees
Blee Thumbnail
Artificial Intelligence • Marketing Tech • Software
New York, New York
30 Employees
Vega Thumbnail
Artificial Intelligence • Automotive • Insurance • Transportation
US
43 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account