GitHub · 去对齐工具与仓库·· 20 天前AI 评分38
mini-infer:从零用 PyTorch 编写的轻量 LLM 推理引擎
ysaatci/mini-infer
来源与原文
GitHub · 去对齐工具与仓库
发布时间:(北京时间)
阅读原始来源 · github.com ↗完整链接与内容说明
https://github.com/ysaatci/mini-infer
本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。
AI 导读
开源项目 mini-infer 是一个用 PyTorch 从零编写的轻量 LLM 推理引擎,用于服务 Qwen2.5 模型,旨在理解 vLLM 等推理服务器的工作原理而非替代它们。项目逐一实现了 KV cache、Continuous batching、Paged KV cache、Speculative decoding 和 int8 weights 等推理优化,每步附 benchmark。测试环境为 RTX 5050 8 GB、Qwen2.5-1.5B、bf16,可在 Linux 或 WSL2 上以 CUDA 12.8+ 运行。
来源:GitHub · 去对齐工具与仓库 · github.com