跳到正文
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