机器之心 前天 17:17
1200行代码逆袭!DeepSeek工程师开源轻量级vLLM,吞吐量逼近原版
index_new5.html
../../../zaker_core/zaker_tpl_static/wap/tpl_guoji1.html

 

Nano-vLLM是由DeepSeek AI研究者俞星凯开发的轻量级vLLM实现,旨在提供高性能的LLM推理。该项目代码量仅1200行,核心功能包括快速离线推理、易读代码库和优化套件。基准测试显示,Nano-vLLM的推理速度与vLLM相当,支持Prefix缓存、Torch编译和CUDA计算图等功能。俞星凯曾在DeepSeek参与DeepSeek-V3和DeepSeek-R1的开发,并有丰富的项目经验,包括植物大战僵尸Qt版等。Nano-vLLM为LLM推理提供了一种更轻量级的选择。

💡 Nano-vLLM是一个轻量级的vLLM实现,由DeepSeek AI研究者俞星凯开发,代码量控制在1200行以内。

🚀 Nano-vLLM的核心功能包括:快速的离线推理速度,与vLLM相当;易于阅读的代码库,便于理解和维护;以及Prefix缓存、Torch编译和CUDA计算图等优化套件。

📊 在基准测试中,Nano-vLLM在RTX 4070硬件上,使用Qwen3-0.6B模型,对256个序列进行测试,输入长度在100-1024 tokens之间随机采样,输出长度也在100-1024 tokens之间随机采样。测试结果表明,Nano-vLLM与vLLM的输出token相同,时间略长,推理速度略逊一筹,整体表现相当。

👨‍💻 Nano-vLLM的开发者俞星凯是DeepSeek的深度学习系统工程师,曾参与DeepSeek-V3和DeepSeek-R1的开发。他拥有丰富的项目经验,包括植物大战僵尸Qt版等。

机器之心报道

机器之心编辑部


开源社区的人应该对 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 吗?


© THE END 

转载请联系本公众号获得授权

投稿或寻求报道:liyazhou@jiqizhixin.com


文章原文

Fish AI Reader

Fish AI Reader

AI辅助创作,多种专业模板,深度分析,高质量内容生成。从观点提取到深度思考,FishAI为您提供全方位的创作支持。新版本引入自定义参数,让您的创作更加个性化和精准。

FishAI

FishAI

鱼阅,AI 时代的下一个智能信息助手,助你摆脱信息焦虑

联系邮箱 441953276@qq.com

相关标签

Nano-vLLM vLLM LLM推理 DeepSeek AI
相关文章