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

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

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

机器之心报道

机器之心编辑部

开源社区的人应该对 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“幻觉”给予适度包容

 中国青年报客户端北京3月5日电(中青报·中青网记者 周围围 李华锡)今年全国两会,全国政协委员、360集团创始人周鸿祎关注大模型话题。他认为,DeepSeek通过大模型技术范式的重大突破,让大模型加...

DeepSeek-R2大模型上线在即,当贝AI或率先接入

DeepSeek-R2大模型上线在即,当贝AI或率先接入

4月30日凌晨DeepSeek更新Prover-V2大模型,据多方消息透露,国内AI技术领军企业深度求索(DeepSeek)计划于5月发布新一代旗舰大模型DeepSeek-R2,预计比GPT-4成本下...

雷神科技:AI PC个人智慧体“小雷AI”接入DeepSeek

雷神科技:AI PC个人智慧体“小雷AI”接入DeepSeek

【雷神科技:AI PC个人智慧体“小雷AI”接入DeepSeek】财联社2月11日电,记者从雷神科技获悉,公司旗下AI PC个人智慧体“小雷AI”11日正式接入DeepSeek。公司表示,“小雷AI”...

人工智能赋能职业教育,北京八维集团引入DeepSeek

人工智能赋能职业教育,北京八维集团引入DeepSeek

在科技飞速发展的当下,人工智能(AI)已成为推动各行业变革的核心力量,职业教育领域自然也不例外。北京八维集团作为职业教育的领军者,敏锐洞察到人工智能带来的巨大机遇,近日正式引入先进的大语言模型 Dee...

医生自嘲“天塌了”!病人查DeepSeek后质疑治疗方案,关键还说对了!AI看病靠谱吗?

医生自嘲“天塌了”!病人查DeepSeek后质疑治疗方案,关键还说对了!AI看病靠谱吗?

本文转自【新民晚报】;“天都塌了!病人DeepSeek后质疑我的治疗方案,气得我自己又查了一遍指南,才发现指南更新了……”2月22日,广东一位医学博主“孤芳自赏”的网帖引起网友热议。截至23日14时,...

联想2月25日开始AI PC端侧DeepSeek模型灰度测试

联想2月25日开始AI PC端侧DeepSeek模型灰度测试

联想集团执行副总裁兼中国区总裁刘军透露,2月25日,联想将对天禧AS做大的升级,将在天禧AS里部署DeepSeek 7B的大模型,这是端侧部署这样规模大模型的首家AI PC厂商。此次升级将会使得联想A...