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

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

3个月前 (06-13)Deepseek应用场景417

机器之心报道

机器之心编辑部

开源社区的人应该对 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当老中医?切勿当真

把DeepSeek当老中医?切勿当真

文汇报见习记者 张菲垭最近,互联网上掀起了一阵“DeepSeek问诊”热潮,不少人开始尝试向AI求一张中医药方,而DeepSeek则一瞬间化身“老中医”。足不出户,AI问诊,情况如何?不少网友测试后发...

西部利得基金完成DeepSeek大模型本地化部署

西部利得基金完成DeepSeek大模型本地化部署

证券时报网讯,西部利得基金宣布,公司在数字化建设领域取得重要进展,完成了DeepSeek多个大模型的本地化部署,并搭建了公司内部AI应用平台。这一举措不仅标志着公司在智能化转型上的突破性进展,也为后续...

DeepSeek你不要太爱普洱了

DeepSeek你不要太爱普洱了

今年春节中国AI公司深度求索开发的大模型DeepSeek火爆“出圈”今天让我们以“旅居”角度对话DeepSeek去看看它眼中旅居普洱是什么样的吧DeepSeek笔下的普洱对话DeepSeek解锁旅居普...

市妇联发布《DeepSeek+妇女工作操作指南》

市妇联发布《DeepSeek+妇女工作操作指南》

随着人工智能时代的到来,“DeepSeek+政务”模式逐渐成为热潮,但如何安全高效用好AI工具,平衡好政务服务的“精度”与“温度”,又是一个崭新的难题。5月13日,杭州市妇联正式发布《DeepSeek...

DeepSeek赋能:从90亿跃升6700亿

DeepSeek赋能:从90亿跃升6700亿

原创 湖北应急管理 湖北应急管理近日,湖北省应急管理厅传来振奋消息,风险隐患监测预警平台完成重大技术升级,依托移动云算力支撑,核心AI模型“楚小应”对接“满血版”DeepSeek - R1(671B)...

“DeepSeek医生”也能上岗“诊疗”了!人工智能自动生成处方靠谱吗?

“DeepSeek医生”也能上岗“诊疗”了!人工智能自动生成处方靠谱吗?

据中央广播电视总台中国之声《新闻纵横》报道,最近,人工智能的春风席卷各大领域,医疗行业也在经历着一场深刻的变革。各地多家医院纷纷引入国产AI大模型DeepSeek,并通过本地化部署将其应用于临床、科研...