← Back
SiTech Team⏱️ 3 წთ. საკითხავი

NVIDIA ModelExpress — Distributing AI Models at Light Speed Across GPU Clusters

NVIDIA ModelExpress — Distributing AI Models at Light Speed Across GPU Clusters

NVIDIA introduced ModelExpress — a system that transfers AI model weights from GPU to GPU over P2P RDMA. DeepSeek-V4 Pro startup time dropped from 8 minutes to 1:44.

NVIDIA ModelExpress — A New Era for Model Distribution

As AI models grow larger, efficiently distributing their weights becomes increasingly critical. Models like DeepSeek-V4 Pro occupy hundreds of gigabytes, and loading them onto GPU clusters takes minutes. The core idea is deceptively simple: before loading a model, first ask where a compatible copy already lives.

How ModelExpress Works

When a serving peer already holds compatible weights in GPU, MX transfers them directly from GPU to GPU over P2P RDMA via NVIDIA Inference Xfer Library (NIXL), bypassing redundant access to object storage, local disk, and host memory. When no peer is available, MX bootstraps from the fastest supported path by streaming from an object store without landing on disk.

MX transfers DeepSeek-V4 Pro weights and JIT Kernel cache artifacts from a serving replica into a fresh replica in under 10 seconds, reducing total startup time from 8 minutes to 1 minute 44 seconds.

Three Paths to GPU Memory

Every new worker must get its weights from one of three places: remote storage (HF or S3), local storage, or another worker already serving the model. For the first worker, there is no peer yet, so it must bootstrap from storage. MX can stream checkpoints from object storage or load from fast local storage, removing avoidable copies along either path.

Peer-to-Peer RDMA: The Fastest Path

Once that first worker is serving, the preferred source changes. Its weights are already resident, post-processed, and laid out in GPU memory. Every compatible worker after it should load directly from that peer over P2P RDMA. MX makes this transition automatically: bootstrap once from storage, then scale out GPU to GPU, falling back to storage only when no compatible peer is available.

Reinforcement Learning and JIT Cache Reuse

MX extends the same approach to RL post-training workflows. As policy weights are continuously updated during RL training, MX efficiently distributes updated weights from trainers to rollout workers. MX also inherits JIT kernel caches from existing replicas, eliminating costly recompilation.

Control Plane and Discovery

MX uses Redis or Kubernetes metadata to discover compatible weight sources across the cluster. A control plane probes available sources along a priority chain: P2P RDMA peer first, then object storage, then local storage via GDS.

Conclusion

NVIDIA ModelExpress solves a fundamental infrastructure problem: the time spent moving model weights before useful work can begin. By reducing startup from 8 minutes to under 2 minutes, it makes AI infrastructure more efficient and responsive. For Georgian developers running AI workloads, this means faster deployments and lower costs.

📖 Source