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

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

10个月前 (06-13)Deepseek应用场景678

机器之心报道

机器之心编辑部

开源社区的人应该对 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大模型接入

日前,市人民政府网站完成DeepSeek大模型接入工作,同步梳理并接入全市政府网站群近153万条海量数据。自上线以来,DeepSeek大模型智能问答累计处理各类咨询、办事等业务请求超1.5万件次,政民...

平安健康成功部署DeepSeek,AI赋能医疗领域再升级

平安健康成功部署DeepSeek,AI赋能医疗领域再升级

近日,平安健康宣布,其已成功完成DeepSeek这一国产AI通用大模型的部署,并在部分应用场景中进行了验证。这一举措标志着平安健康在降低AI部署门槛的同时,正快速推进AI技术在医疗领域的广泛应用。De...

积极拥抱AI技术,广东博众接入DeepSeek大模型

积极拥抱AI技术,广东博众接入DeepSeek大模型

近年来,人工智能已成为推动各行业变革的核心力量。从金融领域的智能投顾到医疗行业的疾病诊断,从教育行业的个性化学习到交通领域的自动驾驶,AI技术正以前所未有的速度重塑着我们的生活和工作方式。在这场技术革...

杭州文旅智能体“杭小忆”全面接入DeepSeek!旅游体验开启“超进化”时代

杭州文旅智能体“杭小忆”全面接入DeepSeek!旅游体验开启“超进化”时代

潮新闻客户端 记者 杨希林3月1日,全国首个城市级文旅智能体杭小忆,全面接入DeepSeek-R1。此次升级,杭州城市大脑智能中枢以“隐形桥梁”之姿,实现DeepSeek-R1在政务信创环境的部署应用...

DeepSeek炒股,靠不靠谱?

DeepSeek炒股,靠不靠谱?

“当没有足够信息的时候,大模型会根据用户要求,自己‘脑补’信息,作出回答,并非基于真实世界的逻辑和事实。”文 / 巴九灵最近小巴在后台看到一条留言:“我今年60多岁了,想用DeepSeek来帮我炒股,...

云南电信携手DeepSeek赋能数字云南建设

云南电信携手DeepSeek赋能数字云南建设

今年以来,中国电信云南公司与DeepSeek开展合作,积极探索AI大模型技术在本地化应用方面的实践。双方在教育教学、基层社会治理、政务服务等多个领域进行了有益尝试,初步形成了一些创新应用案例,为行业数...