InferNEON 用裸机 Rust 内核把 MoE 模型塞进廉价开发板

paraschopra · x · 2026-10-27

paraschopra 的 residency 学员 Karthik 在做 InferNEON,探索大语言模型能被压到多便宜的嵌入式硬件上。MoE 模型每个 token 只激活少数专家,但全部专家远超单板内存,多数边缘方案只能从闪存分页加载、速度极慢。他从三方面入手:一是无 OS、无 libc 的裸机 Rust 推理内核(nostd + 手写 ARM NEON),已在 QEMU 中逐 token 对齐运行 OLMoE-1B-7B;二是用真实 router trace 与回放模拟器测量专家缓存/预测何时有用——目前是原始容量在慢存储上大幅胜过聪明的缓存预测;三是「eNPU」:一块以太网连接的廉价加速板,专家以三值形式常驻、只有激活值过线,在树莓派 CM4 上三值化让专家从访存受限变为计算受限(约比 bf16 快 4.5 倍),每层一次往返约 0.25ms。长期目标:把多块廉价板当作一个内存与算力池,让「一顿饭钱」的硬件跑起严肃的本地 AI。

原文链接 →

「编程与Agent」频道最新

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