
pd-bridge
chadhurley25075-png
Heterogeneous prefill/decode for DeepSeek-V4-Flash: CUDA prefill (DGX Spark, vLLM) -> Metal decode (Mac Studio, oMLX) over plain 10GbE
AI 简介
pd-bridge 是一个支持异构预填充与解码的 LLM 推理桥接系统,实现 CUDA 加速的 vLLM 预填充(运行于 NVIDIA DGX Spark)与 Metal 加速的 oMLX 解码(运行于 Apple M3 Ultra Mac Studio)在普通 10GbE 网络上的协同推理。核心特点是跨框架、跨硬件、跨量化格式(FP8 vs MXFP4)的 KV 缓存序列化与传输,无需 RDMA 或专用互联,保持解码吞吐稳定(23–25 tok/s),显著缩短超长提示(最高 100 万+ token)端到端延迟。适用于需兼顾大规模预填充算力与低功耗/本地化解码的混合部署场景,如科研级超长上下文服务、边缘协同推理原型验证。
Python
Apache License 2.0109
Stars
8
Forks
102
Watchers
0
Issues
Star 增长
今日0
近 7 天0
近 30 天+7
综合评分43.56
默认分支main