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

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

2周前 (06-13)Deepseek应用场景129

机器之心报道

机器之心编辑部

开源社区的人应该对 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,吞吐量逼近原版” 的相关文章

鸿合科技:自研AI平台已接入DeepSeek开源大模型应用于教育场景

鸿合科技:自研AI平台已接入DeepSeek开源大模型应用于教育场景

金融界4月9日消息,有投资者在互动平台向鸿合科技提问:公司有没有对deepseek适配和接入?公司回答表示:公司自主研发AI应用开放平台,并积极推动AI技术在教育场景的应用。目前平台已接入DeepSe...

deepseek怎么样?与豆包相比呢?

deepseek怎么样?与豆包相比呢?

我认为deepseek和豆包两个AI软件没有可比性,主要是定位不同,有各自的特点1️⃣比如deepseek偏推理性,能够把自主思考和推理的过程展示出来,你会发现其实自己有时候陷在自有思维里面 dee...

华为+DeepSeek,终于不再“服务器繁忙”?

华为+DeepSeek,终于不再“服务器繁忙”?

没有人不在期待大模型能够成为下一个电动车,作为代表中国的新兴产业,在世界范围内掀起狂澜。然而主流的MoE架构大模型,却苦于其结构上的“先天不足”:巨大的硬件成本与多重拖累效率的环节,使得中国企业在这场...

多家能源企业接入DeepSeek,能源行业迎来数智化跃迁

多家能源企业接入DeepSeek,能源行业迎来数智化跃迁

近期,多家能源公司纷纷接入DeepSeek大模型。2月15日,中国华能集团有限公司完成了DeepSeek系列模型的本地化部署,推出了“睿智小能”AI助手,与“iHN+”移动门户实现集成,为日常办公与管...

华阳集团:DeepSeek部署有效提升研发效率和生产力

华阳集团:DeepSeek部署有效提升研发效率和生产力

中证报中证网讯(王珞)2月18日,华阳集团微信公众号发文表示,公司自研的软件开放平台已接入DeepSeek的API接口,公司的研发系统已初步完成了对DeepSeek的部署,有效提升了研发效率和生产力。...

科技平权与文化普惠的时代浪潮,DeepSeek与珠江桥牌的时代共鸣

科技平权与文化普惠的时代浪潮,DeepSeek与珠江桥牌的时代共鸣

当国产AI公司DeepSeek以技术普惠的雄姿风靡全球科技圈之际,珠江桥牌提出一酱做成广东招牌菜的新式产品,巧妙地化解着现代生活的种种困境。就像美国博主“吃货老外铁蛋儿”,他用一口地道的大连方言,生动...