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

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

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

机器之心报道

机器之心编辑部

开源社区的人应该对 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 推动金融服务从“信息化”迈向“认知化”

21世纪经济报道记者 张欣 北京报道3月8日,工商银行宣布在同业中率先完成 DeepSeek 最新开源大模型的私有化部署,并将其接入行内“工银智涌”大模型矩阵体系,有力推动金融业务场景的智能化升级。据...

关于DeepSeek V3/R1 Decoding吞吐极限的估计

关于DeepSeek V3/R1 Decoding吞吐极限的估计

经历了一周DeepSeek的打脸活动后,周六 DeepSeek终于开大放出来了自身的推理系统水平,DeepSeek:DeepSeek-V3 / R1 推理系统概览 。在这个结论放出来之前,没...

联想与沐曦DeepSeek一体机上市首月发货量突破千台

联想与沐曦DeepSeek一体机上市首月发货量突破千台

【联想与沐曦DeepSeek一体机上市首月发货量突破千台】《科创板日报》7日讯,《科创板日报》记者获悉,联想集团与沐曦股份合作的首个国产DeepSeek一体机解决方案发布一个月以来,截至今日,其累计发...

DeepSeek+水利 治水迎“智”变

DeepSeek+水利 治水迎“智”变

今年以来,随着以DeepSeek为代表的人工智能技术向垂直领域加速渗透,水利行业也迎来新一轮技术应用变革。目前,全省水利系统正积极开展“DeepSeek+水利”的相关应用,除了已经上线“DS都江堰”智...

ST广网:公司“秦岭云”平台已完成DeepSeek本地化部署

ST广网:公司“秦岭云”平台已完成DeepSeek本地化部署

每经AI快讯,有投资者在投资者互动平台提问:董秘您好!请问贵公司是否已经部署了DeepSeek?如果已经部署了,请问主要应用于哪些具体的业务?公司接入DeepSeek有哪些成本、收益方面的考量?如果公...

一心堂:公司已完成DeepSeek-R1在部分场景的接入

一心堂:公司已完成DeepSeek-R1在部分场景的接入

 同花顺(300033)金融研究中心04月29日讯,有投资者向一心堂(002727)提问, 有否考虑线上接入DeepSeek-R1,向业务、员工、加盟商及合作伙伴的智能体,包括提供即时、准确的问答支持...