当前位置:首页 > Deepseek应用场景 > 正文内容

1200行代码逆袭!DeepSeek工程师开源轻量级vLLM,吞吐量逼近原版

5个月前 (06-13)Deepseek应用场景526

机器之心报道

机器之心编辑部

开源社区的人应该对 vLLM 不陌生,它是一个由加州大学伯克利分校团队开发的高性能、开源 LLM 推理和服务引擎,核心目标是提升 LLM 的推理速度(吞吐量)和资源利用率(尤其是内存),同时兼容 Hugging Face 等流行模型库。

简单来说,vLLM 能让 GPT、Mistral、LLaMA 等主流模型系列跑得更快、消耗更少资源,取得这些效果的关键是其创新的注意力机制实现方案 ——PagedAttention。

近日,DeepSeek AI 研究者、深度学习系统工程师俞星凯从零开始构建了一个轻量级 vLLM 实现 ——Nano-vLLM,将代码简化到了 1200 行。

目前,该项目在 GitHub 上收获了 200 多的 Star。

GitHub 地址:https://github.com/GeeeekExplorer/nano-vllm/tree/main

具体来讲,Nano-vLLM 具有以下三个核心功能:

一是,Fase 离线推理。推理速度与 vLLM 相当。

二是,易读代码库。实现非常简洁,Python 代码减少到了 1200 行以下。

三是,优化套件。提供 Prefix 缓存、Torch 编译、CUDA 计算图等功能。

俞星凯在基准测试中采用了以下测试配置:

硬件:RTX 4070

模型:Qwen3-0.6B

总请求:256 个序列

输入长度:100–1024 tokens 之间随机采样

输出长度:100–1024 tokens 之间随机采样

基准测试结果如下表所示,Nano-vLLM 与 vLLM 的输出 token 相同,时间略长,推理速度(吞吐量)稍逊一点点。整体而言,二者表现相当。

作者简介

Nano-vLLM 开发者俞星凯目前就职于 DeepSeek,参与过 DeepSeek-V3 和 DeepSeek-R1 的开发工作。

有意思的是,根据其 GitHub 主页,他还曾开发过一个植物大战僵尸 Qt 版,该项目也已经收获了 270 多星。此外,由于毕业于南京大学,他还曾参与了不少南京大学的计算机项目,包括南京大学计算机图形学绘图系统、南京大学分布式系统 Raft 算法最简实现、南京大学操作系统 OSLab 等。

而根据其 LinkedIn 页面可知,他曾先后在腾讯、幻方(DeepSeek 母公司)和字节跳动实习过。2023 年后入职 DeepSeek 成为深度学习系统工程师。

你是 vLLM 用户吗?会考虑尝试 Nano-vLLM 吗?


“1200行代码逆袭!DeepSeek工程师开源轻量级vLLM,吞吐量逼近原版” 的相关文章

新版DeepSeek-V3接入每经大模型平台!小编亲测:敲代码、写报告、撰长文样样在行

新版DeepSeek-V3接入每经大模型平台!小编亲测:敲代码、写报告、撰长文样样在行

3月24日,DeepSeek悄悄上线了新版V3开源大模型——DeepSeek-V3-0324。专业AI模型评测机构Artificial Analysis最新排名显示,新版DeepSeek-V3在基准测...

继DeepSeek席卷全网之后 Manus再度霸屏

继DeepSeek席卷全网之后 Manus再度霸屏

阅读提示DeepSeek、Manus相继火爆全网,是中国人工智能产业厚积薄发的有力见证,彰显出中国在该领域蕴藏的巨大潜力。AI的“狂飙”给企业发展带来了前所未有的启示:科技创新激起澎拜动能,进而带动全...

华南理工大学上线满血版DeepSeek!面向省内院校开放

华南理工大学上线满血版DeepSeek!面向省内院校开放

好消息!新学期即将开始华南理工大学本地化部署的满血版DeepSeek-R1正式上线!为进一步培养师生AI思维和使用AI的能力华南理工大学基于学校“成务”科学计算平台成功完成满血版DeepSeek-R1...

兆芯全系列整机形态成功部署 DeepSeek R1 大模型

兆芯全系列整机形态成功部署 DeepSeek R1 大模型

IT之家 2 月 24 日消息,上海兆芯集成电路股份有限公司今日发文宣布:近日,基于兆芯高性能通用处理器的 PC 终端、工作站以及服务器成功实现 DeepSeek-R1-Distill 模型(1.5B...

中新天津生态城“城市大脑”全面接入DeepSeek

中新天津生态城“城市大脑”全面接入DeepSeek

人民网天津3月27日电 (记者孙翼飞)近日,中新天津生态城(下简称“生态城”)“城市大脑”完成DeepSeek(深度求索)大模型的私有化部署,实现与城市大脑的初步集成,标志着城市大脑智慧服务能力正式迈...

赵满意:DeepSeek浪潮下AI大模型赋能产业发展

赵满意:DeepSeek浪潮下AI大模型赋能产业发展

5月10日,由中国企业改革与发展研究会主办,中企研数字经济与数据资产工作委员会、企业观察报社承办的央企AI+大模型应用论坛在北京举办。论坛上,华为技术有限公司北京昇腾创新中心负责人赵满意以《DeepS...