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

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

3周前 (06-13)Deepseek应用场景159

机器之心报道

机器之心编辑部

开源社区的人应该对 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,模型与芯片深度协同

IT之家 5 月 8 日消息,在今日盛大举行的鸿蒙电脑技术与生态沟通会上,华为首款鸿蒙电脑正式揭开神秘面纱,惊艳亮相。这一消息瞬间引发了科技圈的广泛关注,众多业内人士与科技爱好者都对这款具有开创性意义...

“卖的就是信息差”!借DeepSeek做社群月入18万?“教程”标价10万……紧急声明

“卖的就是信息差”!借DeepSeek做社群月入18万?“教程”标价10万……紧急声明

某热门DeepSeek社区会员费为61元,总会员数超过3600人,在DeepSeek爆红的近一个月里,仅会员费收入就超过18万元。今年春节,国产AI公司深度求索开发的大模型DeepSeek成为爆款。据...

同程旅行宣布接入DeepSeek大模型,打造一站式智能旅行服务

同程旅行宣布接入DeepSeek大模型,打造一站式智能旅行服务

3月3日消息,同程旅行日前宣布其自主研发的旅游行业专属模型——同程程心大模型将全面接入DeepSeek,成为首家接入DeepSeek大模型的在线旅游平台(OTA)。目前,该功能已进入内测阶段,预计3月...

上海交通大学鲲鹏昇腾科教创新卓越中心特训营DeepSeek专场圆满举办

上海交通大学鲲鹏昇腾科教创新卓越中心特训营DeepSeek专场圆满举办

央广网北京3月28日消息 3月26日下午,上海交通大学鲲鹏昇腾科教创新卓越中心特训营-DeepSeek专场在上海交通大学闵行校区图书信息楼成功举办。活动旨在基于昇腾技术的DeepSeek本地化部署实践...

董明珠:将基于 DeepSeek 等打造格力智能家居垂类大模型

董明珠:将基于 DeepSeek 等打造格力智能家居垂类大模型

IT之家 3 月 10 日消息,据证券日报昨日晚间报道,格力董明珠透露:“格力电器已成功将自有的多模型框架与 DeepSeek R1 满血版大模型深度融合。”“例如,这一合作成果已经体现在格力电器的语...

首次接入DeepSeek、7*24小时在线 企业登记注册积累“北京经验”

首次接入DeepSeek、7*24小时在线 企业登记注册积累“北京经验”

大模型技术的接入让企业登记注册更加便捷。3月4日,北京商报记者从北京市市场监督管理局获悉,北京市市场监管局企业登记注册场景已正式接入DeepSeek大模型技术,实现企业登记注册业务24小时在线“智能咨...