#387 · 主分类: 教育与研究

tiny-vllm

ai attention batching course cpp cuda hpc inference llm llm-inference pagedattention tiny-vllm vllm

用C++和CUDA构建你自己的高性能LLM推理引擎——vLLM的缩小版

项目最后更新:08/23/26

GitHub Stars

1.1K

Forks数量

84

贡献者数量

4

许可证

Apache-2.0

收录理由

tiny-vllm 是一份难得的双重礼物:它既是一个用 C++ 和 CUDA 写成的、能真正跑起来的 LLM 推理服务,又是一门手把手带你把它造出来的课程。作者从零开始推导数学原理和实现细节,手写 CUDA 内核覆盖完整的前向传播、KV 缓存、连续批处理和 PagedAttention,学完你手里会有一个可运行的引擎,心里也真正明白它每一步在做什么。这对两类人特别有价值:用过 llama.cpp 或 vLLM 却从没看过内部实现的开发者,可以顺着它看清从加载 Safetensors 模型到生成 token 之间发生的每一件事;而讲师和自学者则能得到一条现成的教学路线,从 bfloat16、RMSNorm 一路讲到 online softmax 和分页 KV 缓存,既适合大学课堂当教材,也适合自己按节奏啃完整个项目。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目