Python · 项目报告

PipeNetwork/kimi-k3-mlx

MLX port of moonshotai/Kimi-K3 (2.78T multimodal MoE): streaming converter, REAP expert pruning, and per-language expert-overlap analysis

已完成 打开 GitHub
P
316星标
34Fork
15Issue
未知许可证

分析结果

项目分析

该仓库是 moonshotai/Kimi-K3 的 Apple Silicon MLX 移植项目,面向 2.78T 总参数、104B 激活参数的原生多模态 MoE 模型。项目提供文本塔、视觉塔、mlx-vlm 多模态封装、流式权重转换器、MXFP4 量化支持、REAP 专家剪枝版本,以及针对语言/专家重叠的分析能力。README 明确指出完整模型无法在单台 Mac 上运行,实际可用重点在已发布的剪枝版 MLX 权重,例如 350GB/451GB 级别模型,可在 512GiB 级别的 M3 Ultra 上以约 5.5 tok/s 交互式推理。

适用领域 大语言模型推理 / 多模态大模型 / Apple Silicon / MLX 生态 / MoE 专家模型 / 模型量化 / 模型剪枝 / Kimi-K3 / Moonshot AI 模型适配 / 本地 AI 部署 / 中文与代码场景大模型
配置难度 高。该项目不是普通 Python 应用或轻量模型部署项目,而是超大 MoE 多模态模型在 MLX 生态下的底层移植和运行工程。只做现成剪枝版推理也需要高端 Apple Silicon、数百 GB 模型管理和 MLX 经验;若要自行转换、调量化、改剪枝或集成多模态 wrapper,则需要深入理解模型架构、权重格式、MLX 张量行为和大模型推理内存瓶颈。
商业价值 对具备高端 Apple Silicon 设备且希望本地运行 Kimi-K3 级别模型的团队有较高价值,尤其适合中文、代码和多模态私有化原型验证。它可以帮助企业或研究团队避免依赖云端 API,探索本地超大 MoE 模型推理、量化和剪枝。但其商业落地受限于硬件成本、模型许可证、部署复杂度和推理速度;更适合高价值低并发场景、研究验证、私有数据实验和技术储备,而不是大规模低成本生产服务。
01

技术亮点

  • 提供 Kimi-K3 在 MLX 上的较完整移植,包括文本塔、视觉塔和多模态 glue。
  • 支持 Kimi-K3 特有结构:SiTU-GLU、AttnRes、LatentMoE、q-LoRA + gated MLA、per-channel KDA decay。
  • 实现流式转换器,解决 5.6TB bf16 模型无法整体物化转换的问题。
  • 对 MXFP4 源专家权重进行 bit-exact 转换,避免普通 4-bit 二次量化带来的误差和体积劣化。
  • 视觉塔经过与 Moonshot torch 实现端到端 parity 测试,README 声称相对误差可达 1.5e-6。
  • 多模态 wrapper 特别处理 K3 的 <|media_pad|> 扩展机制,避免模型看似正常但实际丢失大部分图像 token。
  • 提供多个已发布剪枝版本,其中 REAP80 约 350GB,REAP73 约 451GB,并给出在 512GiB M3 Ultra 上约 5.5 tok/s 的参考性能。
  • 包含中文+代码校准版本 Kimi-K3-REAP73-zh-code-MLX-mxfp4-q8,对中国开发者更有实用价值。
  • README 对架构、参数预算、量化 tier、硬件现实约束说明非常详细,工程透明度高。
02

目标用户

  • 使用 Apple Silicon Mac Studio/Mac Pro/M3 Ultra 等设备进行本地大模型推理的开发者
  • 研究 Kimi-K3、MoE、KDA、MLA、AttnRes 等架构细节的模型工程师
  • 希望在 MLX 或 mlx-lm/mlx-vlm 生态中运行超大模型的开发者
  • 关注中文、代码、多模态能力的 AI 应用开发者
  • 需要研究专家剪枝、MXFP4 量化、模型转换流程的算法和系统研究人员
  • 有 TB 级模型下载、存储和内存预算的高级用户
03

配置要求

  • 硬件:完整 1.56TB mxfp4 模型无法在单台 Mac 上运行;已发布剪枝模型仍需要 350GB 到 451GB 级别磁盘空间,实际运行推荐 512GiB 统一内存的 M3 Ultra 级设备。
  • 操作系统:Apple Silicon macOS 环境,依赖 MLX。
  • Python:需要 Python 与 MLX 生态兼容版本。
  • 核心依赖:mlx、mlx-lm、mlx-vlm、numpy、torch、Pillow、Hugging Face 相关工具等,具体版本需参考仓库依赖文件。
  • 模型权重:需要从 Moonshot 原始 Kimi-K3 或作者发布的 Hugging Face MLX artifact 获取权重。
  • 量化要求:源专家权重是 MXFP4,应使用 MLX 原生 mxfp4 路径;不能简单按 compressed-tensors 映射为普通 affine 4-bit,否则会错误解读权重。
  • 推理配置:已发布剪枝模型中 --nonexpert-bits 8 是关键配置之一,否则部分构建可能超过 512GiB 内存并被 OOM kill。
  • 多模态配置:需要正确处理 <|media_pad|> 扩展逻辑,一个图片占位符必须扩展为完整图片特征块,不能使用普通 LLaVA/Kimi-VL 的 same-length scatter。
04

适用场景

  • 在高内存 Apple Silicon 设备上本地运行 Kimi-K3 剪枝版模型
  • 将 Moonshot Kimi-K3 权重转换为 MLX 可加载格式
  • 验证 Kimi-K3 文本塔、视觉塔和多模态 wrapper 的结构正确性
  • 研究 MXFP4 源权重到 MLX mxfp4 的无损转换
  • 基于 REAP 专家剪枝减少模型存储和内存占用
  • 运行图文输入的多模态推理,例如图片描述、视觉问答
  • 分析不同语言或任务下的 MoE 专家重叠情况
  • 为中文和代码任务选择或构建特化剪枝模型
05

部署与配置

  • 准备 Apple Silicon 环境,推荐高内存 M 系列设备;完整模型不适合单机部署,剪枝版也通常需要 350GB 到 451GB 以上存储和接近 512GiB 统一内存。
  • 安装 Python 环境,建议使用虚拟环境或 Conda。
  • 安装 MLX 相关依赖,例如 mlx、mlx-lm、mlx-vlm,以及项目需要的 Python 包。具体依赖需以仓库 requirements 或 pyproject 为准。
  • 克隆仓库:git clone https://github.com/PipeNetwork/kimi-k3-mlx.git
  • 如只想推理,优先从 Hugging Face 下载作者发布的剪枝版 MLX 权重,例如 Kimi-K3-REAP80-MLX-mxfp4-q8、Kimi-K3-REAP73-MLX-mxfp4-q8 或 Kimi-K3-REAP73-zh-code-MLX-mxfp4-q8。
  • 如需自行转换原始 Kimi-K3 权重,使用 scripts/convert.py,而不是 mlx_lm convert,因为 K3 bf16 权重约 5.6TB,普通转换会物化整个模型导致不可行。
  • 根据 README 中的脚本运行文本或多模态推理,例如使用 scripts/vl_generate.py 进行图文推理验证。
  • 运行测试脚本验证转换和结构正确性,例如 tests/test_kimi_k3.py、tests/test_vision_parity.py、tests/test_vl_wrapper.py、tests/test_convert_roundtrip.py。
06

风险与注意事项

  • 硬件门槛极高:即使剪枝版也通常需要数百 GB 存储和接近 512GiB 统一内存,普通 MacBook 或低内存 Mac 无法使用。
  • 完整模型单机不可运行,README 明确说明最小完整 tier 约 870GB,超过当前最大 Apple Silicon 单机可用上限。
  • 项目复杂度高,涉及 Kimi-K3 专有结构、MXFP4 格式、MoE 专家剪枝、多模态 token 扩展,修改或二次开发容易引入静默错误。
  • 普通 mlx_lm convert 不适用,错误转换可能导致内存爆炸或权重格式错误。
  • 如果误将 compressed-tensors 当作普通 affine 4-bit 处理,会产生错误模型。
  • 剪枝版本损失的信息来自专家裁剪,虽然可运行,但质量可能与完整 Kimi-K3 有差距,需按业务场景自行评估。
  • 当前仓库 license 字段为空,商业使用前需要核查仓库代码许可证、Moonshot Kimi-K3 权重许可证以及 Hugging Face artifact 的使用条款。
  • 多模态推理对 processor、token placeholder、图像 token 展开要求严格,集成到其他框架时风险较高。
  • 下载、存储和加载成本巨大,CI/CD、容器化、云同步和备份都不友好。

历史记录

热榜历史快照

2026-08-03 第25名 新收录 · github_search
2026-08-02 第22名 新收录 · github_search
2026-08-01 第21名 新收录 · github_search
2026-07-31 第26名 新收录 · github_search