Devotd Github
新发现推理与服务

haochenglu/pi-attn — 兼顾优化与可复现性的推理引擎

haochenglu/pi-attn 是一个基于 Rust 的语言模型服务引擎,通过固定运算顺序,在批处理和缓存的情况下也能保持结果的可复现性。

仓库信息haochenglu/pi-attn ↗

haochenglu/pi-attn 是一个基于 Rust 的语言模型服务引擎,通过固定运算顺序,在批处理和缓存的情况下也能保持结果的可复现性。

GitHub 项目说明: Bitwise-reproducible LLM serving with every optimization on, at full speed. Paper, engine, kernels and results.

星标
★ 3
复刻
0
语言
Rust
查看原始来源 ↗在 GitHub 查看 ↗

haochenglu/pi-attn 是一个语言模型服务引擎,目标是在启用批处理、前缀缓存和分块预填充的情况下也能复现结果。仓库说明指出,它以在开启优化的同时实现位级可复现性为目标。由于没有最新的发布信息,无法指向任何特定版本的发布内容或变更。这是基于随仓库名称一并公开的项目说明、值得进一步考察的一项发现。

位级可复现性是指在相同输入下计算结果逐位一致的性质。PI-Attn 通过固定浮点运算顺序进行设计,使批处理或缓存不会影响结果。同时,其运算拆分方式的组织也与快速内核的形态相匹配。仓库中包含 Rust 执行引擎和注意力(attention)运算内核,还包括训练器复现验证工具和服务器兼容性检查工具。论文、结果文件和自动评估脚本一并提供,将可用于考察性能与可复现性主张的材料集中在一处。

不过,公开信息中并未记载正式发布版本与许可证。因此,无法确认固定的分发版本或使用许可条件。关于性能和可复现性的判断,也难以超出仓库所包含的论文、结果和评估工具的范围。这是一个处于早期阶段的项目,引入前应审查相关资料并确认分发条件。