工程师用 C 重写 Transformer 引擎,发现难点在周边系统

RelevantShape3963 · reddit · 2026-07-22

一位固件工程师用 C 从零重写 Transformer 引擎

作者是一名固件工程师,从 2022 年开始利用午休时间,用 18 个月 写出一个 约 1.5 万行 的 C 语言 transformer engine,能够跑 Gemma、Llama、GPT-2、PaliGemma。

他最大的收获是:引擎只占一半工作量。真正难的是把 tokenizer、chat template、KV cache 等周边“管线”接起来,让模型能稳定可用。

另外一个有意思的观察是:当模型 每几秒才吐一个 token 时,反而更容易看清它在做什么。这个项目后来被整理成一本“field diary”式的书,作者还开源了引擎和免费章节。

原文链接 →

「编程与Agent」频道最新

更多「编程与Agent」频道 AI 资讯 →