跳到正文
GitHub · 灵魂体仓库·· 20 天前精选AI 评分59

nanoinfer:从零手写的 Qwen2.5-0.5B 推理引擎开源

SpookyJumpyBeans/nanoinfer

来源与原文

GitHub · 灵魂体仓库

发布时间:(北京时间)

阅读原始来源 · github.com ↗
完整链接与内容说明

https://github.com/SpookyJumpyBeans/nanoinfer

本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。

AI 导读

SpookyJumpyBeans 在 GitHub 发布 nanoinfer,一个从零编写的 LLM 推理引擎,目标模型为 Qwen2.5-0.5B-Instruct,直接加载原始 .safetensors,运行时仅依赖 numpy,tokenizer、前向传播、KV cache、采样和量化均为自行实现。分八个阶段对照 transformers 和 llama.cpp 逐项验证,包括 KV cache 带来 3.5 倍加速、INT8 量化仅 +0.17% 困惑度、Rust AVX2 内核比 OpenBLAS 快 22%,以及 f32 解码与 llama.cpp 基本持平;并配有可视化页面展示每步候选 token 和注意力。

推荐理由

作者从零实现了一个推理引擎并用逐阶段基准量化每一步的收益,读者可以从中借鉴带宽瓶颈、量化和内核优化的可迁移测量方法。

来源:GitHub · 灵魂体仓库 · github.com