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

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

2个月前 (06-13)Deepseek应用场景267

机器之心报道

机器之心编辑部

开源社区的人应该对 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赋能,美的空调开启 AI 智能新篇

DeepSeek赋能,美的空调开启 AI 智能新篇

2025年初,中国人工智能大模型 DeepSeek火遍全球,让全世界看到了中国AI技术的巨大潜力。国内家电行业也迅速乘上了这股东风,持续探索人工智能大模型与智能家居结合的无限可能性。据悉,家电巨头美的...

黄仁勋如何抗过DeepSeek冲击?

黄仁勋如何抗过DeepSeek冲击?

摘要:除了核心业务外,英伟达几乎不为任何事分心。4月13日,华尔街日报发文,人工智能基础设施领域的王者英伟达似乎无处不在,人们很容易将成功原因归结为该公司掌握的强大技术。但它广泛的影响力也反映出一种商...

中国电信浙江公司+DeepSeek开启AI之旅 拥抱智能未来

中国电信浙江公司+DeepSeek开启AI之旅 拥抱智能未来

如今,大模型已成为推动各行业智能化转型的核心驱动力。依托“天翼云”国家级云平台,中国电信浙江公司全面整合DeepSeek开源模型,利用息壤智算一体机-DeepSeek版,为客户提供安全、高效以及本地化...

外界热议:DeepSeek又变强了!

外界热议:DeepSeek又变强了!

中国人工智能初创公司深度求索(DeepSeek)24日深夜低调上线了DeepSeek-V3的新版本DeepSeek-V3-0324,参数量为6850亿,在代码、数学、推理等多个方面的能力再次显著提升,...

DeepSeek与财富管理:慧眼识机,决胜千里

DeepSeek与财富管理:慧眼识机,决胜千里

在数字化浪潮席卷各行各业的今天,传统的财富管理模式正面临前所未有的挑战。财富管理不仅需要处理大量复杂的信息和数据,还需面对市场波动的风险和客户需求的多样化。然而,随着人工智能技术的飞速发展,尤其是De...

DeepSeek“引爆”博鳌!AI将像水电一样人人可用

DeepSeek“引爆”博鳌!AI将像水电一样人人可用

当时间步入博鳌亚洲论坛2025年年会第二天,这场盛会迎来议程最繁忙的阶段,人工智能毫无悬念地成为中外嘉宾瞩目的“C位”。从现场令人惊叹的科技展示,到分论坛上唇枪舌剑的思想交锋,AI热度持续飙升。有嘉宾...