GitHub · 灵魂体仓库·· 21 天前精选AI 评分70
MeKaustubh07 开源从零编写的 Metal 推理引擎,在 8GB MacBook 上跑 Qwen3.5-2B INT4
MeKaustubh07/metal-inference-engine
来源与原文
GitHub · 灵魂体仓库
发布时间:(北京时间)
阅读原始来源 · github.com ↗完整链接与内容说明
https://github.com/MeKaustubh07/metal-inference-engine
本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。
AI 导读
MeKaustubh07 在 GitHub 开源了一个从第一性原理用 Python 编写的 LLM 推理引擎,预填充用 PyTorch GEMM、解码用手写 Metal kernel,在 Apple M2 8GB 上以 INT4 通过 OpenAI 兼容 API 服务 Qwen3.5-2B,支持连续批处理、paged KV cache 和打包分块预填充。基准显示单流解码 50 tok/s、batch-8 聚合解码 116 tok/s、HTTP 8 客户端 94 tok/s(TTFT p50 0.84 秒),对比 llama.cpp Q4_1 与 MLX-LM 4-bit,并与 HF transformers 校验精度(INT4 KL 0.045)。
推荐理由
作者从零实现并给出与 HF、llama.cpp、MLX-LM 的逐项对比和精度数据,方法与权衡都可复查。
来源:GitHub · 灵魂体仓库 · github.com