AWS 机器学习· Sunita Nadampalli·· 3 小时前精选AI 评分63
vLLM 引入推式 KV 缓存传输优化分离式推理 TTFT 与 TPOT
Push-based KV transfer to improve TTFT and TPOT for disaggregated inference in vLLM
AI 导读
AWS 团队向开源推理引擎 vLLM 贡献了推式 NIXL 连接器,通过提前注册目标内存块并在后台线程执行 KV 传输,将传输协调耗时移出分离式推理的关键路径。
推荐理由
文章详细拆解了分离式推理中推式KV传输的设计细节与线程重叠模型,为大模型推理服务降低首字延迟提供了落地工程参考。
来源:AWS 机器学习 · aws.amazon.com