把大模型跑起来,有点像经营一家忙碌的餐厅:既要安排订单,又要腾出操作台,还要避免重复备料。负责这些工作的就是推理引擎——它不训练模型,而是管理显存、调度请求并提高计算吞吐。nano-vLLM 从头实现了一套轻量版本,目的不是替代生产级 vLLM,而是把它复杂的内部机制变成更容易读懂的代码。
据项目官方 README,nano-vLLM 的核心实现约 1200 行 Python,并保留了几项关键优化。前缀缓存会复用多个请求中相同提示词的计算结果,像把常用配料提前备好;张量并行则把同一层计算拆给多块 GPU 协作。项目还实现了 Torch compilation 和 CUDA Graph,提供仿照 vLLM 的离线推理接口,可使用 Hugging Face 上的 Qwen 模型权重。
作者在 RTX 4070 Laptop、Qwen3-0.6B 和 256 个请求的测试中报告,nano-vLLM 吞吐量为 1434.13 tokens/s,vLLM 为 1361.84 tokens/s。这个单一配置的自测不足以证明普遍性能优势,但项目真正值得看的地方,是用较小代码体量展示生产级推理引擎的核心思路。