蚁工厂
26-05-30 14:49 微博认证:科技博主

教学项目tiny-vllm,教你使用 C++ 和 CUDA 构建一个高性能 LLM 推理引擎
地址:github.com/jmaczan/tiny-vllm
“在这个过程中,我们会学到很多东西,也会犯错,并且会从零开始推导相关思想和数学原理。

这个仓库包含两部分内容:
1. 推理服务器的完整源代码
2. 一门课程,我会带你完成这个引擎的实现过程

你可以把它当作学习工具,用在自己的学习路径中。如果你是讲师,也欢迎把它作为大学教学资源。

这个推理引擎包含:

从 Safetensors 加载真实的 LLM 模型:Llama 3.2 1B Instruct
完整的 LLM forward pass:prefill + decode
所有计算都使用 CUDA kernels
KV cache
static batching
continuous batching
online softmax,类似 FlashAttention
PagedAttention”

#AI创造营#

发布于 山东