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

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

18小时前Deepseek应用场景29

机器之心报道

机器之心编辑部

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

广东超40所高校完成DeepSeek教育系统本地化部署,有何影响?你的院校AI了吗?丨AI校园奇遇记

广东超40所高校完成DeepSeek教育系统本地化部署,有何影响?你的院校AI了吗?丨AI校园奇遇记

在人工智能浪潮席卷教育领域之际,新学期伊始,广东多所高校宣布完成DeepSeek-R1大模型的教育系统本地化部署,将这一国产AI大模型融入教学、科研、管理与校园服务场景,为师生提供智能化服务,助力教育...

ai.com跳转至DeepSeek

ai.com跳转至DeepSeek

鞭牛士 2月9日消息,近日有网友发现,输入ai.com域名会跳转到DeepSeek官网,以为DeepSeek买了该域名。输入该域名后,跳转至该页面:2023年,OpenAI的ChatGPT风靡全球,a...

天亿马:下属公司翼启AI政务管家已接入DeepSeek

天亿马:下属公司翼启AI政务管家已接入DeepSeek

金融界3月14日消息,有投资者在互动平台向天亿马提问:公司在大语言模型方面已接入百度,阿里等通用大模型进行测试与研发,请问是否也包括最近很火的deepseek?公司回答表示:公司下属公司“广东翼启”开...

DeepSeek开源放大招:FlashMLA让H800算力狂飙!曝光低成本秘笈

DeepSeek开源放大招:FlashMLA让H800算力狂飙!曝光低成本秘笈

【新智元导读】DeepSeek开源周第一天就放大招!FlashMLA强势登场,这是专为英伟达Hopper GPU打造MLA解码内核。注意,DeepSeek训练成本极低的两大关键,一个是MoE,另一个就...

DeepSeek 入选2025 十大年度 IP

DeepSeek 入选2025 十大年度 IP

DeepSeek 入选2025 十大年度 IPIT之家 5 月 12 日消息,由中国版权协会、中国文化娱乐行业协会等单位指导,新浪财经、微博等单位联合主办的 2025 世界 IP 经济发展大会暨全球...

深夜突袭,DeepSeek-Prover-V2加冕数学王者!671B数学推理逆天狂飙

深夜突袭,DeepSeek-Prover-V2加冕数学王者!671B数学推理逆天狂飙

来源:新智元  【新智元导读】就在刚刚,DeepSeek-Prover-V2技术报告也来了!34页论文揭秘了模型的训练核心——递归+强化学习,让数学推理大提升。有人盛赞:DeepSeek已找到通往AG...