Hiring.Camp

Principal Architect, AI Inference Networking - NIXL and Dynamo

Nvidia

·

Yesterday

Location
Shanghai, Shanghai,CN, CN · Shenzhen, Guangdong,CN, CN
Type
Full-time
Department
Engineering
Seniority
Lead
Education
PhD
Source
Eightfold

Description

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

Python

Similar Jobs

30

Principal AI Architect

Directv·El Segundo-CA-2260 E Imperial Hwy, US +1·Remote

2d ago

Principal AI Architect

Analytica·Bethesda, MD

1w ago

Principal AI Architect

SecurityScorecard·Hybrid·Remote, Hybrid

1w ago

Principal AI Architect

CVS Health·Work At Home-Colorado, US·Remote

3w ago

Principal AI Architect

Robotsandpencils·US - Remote·Remote

1mo ago

Principal Architect AI

Anaplan·London, UK +1

1mo ago

Principal AI Architect

Lplfinancial·Fort Mill, Charlotte +2

1mo ago

Principal AI Architect

Wells Fargo·112265-NJ-MetroPark, Iselin +2

3mo ago

Principal AI Architect

Scanhealthplan·Long Beach Office 3800, US·Remote

4mo ago

Principal AI Architect

Hophr·San Francisco, CA·Onsite

5mo ago

Principal AI Architect

Mainspringenergy·Menlo Park, CA·Remote

5mo ago

Principal Architect AI Platform

BMO is·320Canal, US +5·Remote

4w ago

Principal Architect, AI Platform

BMO is·BMOPLACE, Canada

1mo ago

Sr. Principal Architect, AI Solutions

Ssctech·New York, NY +1·Remote, Hybrid

1mo ago

Associate Principal AI Architect (Claude AI)

Unisys·Bangalore - RGA Tech Park, India

1mo ago

ASIC Architect, Principal (AI Inference)

Positron·Remote +1·Remote

2mo ago

Principal AI Architect - Lifescience

Uipath·Remote-Texas +3·Remote

2mo ago

Principal AI Architect – Healthcare

Uipath·Remote-Texas +4·Remote

2mo ago

Principal Architect AI Data Engineer

EXL Talent Acquisition Team·Gurugram, Haryana·Hybrid

2mo ago

Principal Architect AI Data Engineer

EXL·Gurugram, Haryana·Hybrid

2mo ago

Principal AI Architect for Business Systems

Trimble·US - OH, Dayton +1

3mo ago

Principal AI Architect for Business Systems

Trimble·Dayton, OH +1

3mo ago

Principal AI Architect Knoxville, Tennessee, United States

C2 Labs, Inc·Remote

3mo ago

Senior Principal, AI Architect

Referral Publicisgroupe·Bengaluru, KA

3mo ago

Senior Principal AI Architect/Engineer

Pepsi Co·Plano, TX

4mo ago

Principal Architect, AI/ML

Zencore·Remote·Remote

5mo ago

Principal Architect: AI & GCP Agentic Stack

Cnx·Work-at-Home, TN

6mo ago

AI Strategy, Emerging Systems & AI Principal Architect

Micron Technology·Boise, ID·Hybrid

7mo ago

AI Strategy, Emerging Systems & AI Principal Architect

Micron·Boise, ID - Main Site·Hybrid

7mo ago

Principal Architect, AI/ML

Zencore·EMEA·Remote

7mo ago