本地 LLM 的规模化挑战:模型增长推动分布式需求
deep_candidateTatiana(MLX 团队研究科学家)开场:模型持续变大、能力持续增强。单机已无法满足大模型推理和训练需求,分布式成为必需。
正在接入 Apple HLS 视频源…
Tatiana(MLX 团队研究科学家)开场:模型持续变大、能力持续增强。单机已无法满足大模型推理和训练需求,分布式成为必需。
以 4 台 M3 Ultra 搭建集群为例——了解完整堆栈后将其组合为协同工作的机器组。hostfile 配置包含节点地址和 Thunderbolt 连接信息。
JACCL(MLX 分布式通信库)负责节点间初始协调。RDMA 设备名列表对应每个 Thunderbolt peer 连接。mlx.distributed_config 脚本自动生成配置。
加速比取决于模型大小和架构。分布式不止是提速,有时模型太大单机无法加载(如 1 万亿参数的 Kimi 2.6,即使 8-bit 量化也需要多台机器)。
分布式训练时 batch-size 按设备数缩放,每台机器处理相同样本数。以 Qwen 3.5(9B 参数)为例,对比单机和集群的微调性能。
通过 Thunderbolt RDMA 连接多台 Mac 共享模型权重。JACCL 库处理节点间通信协调。4 台 M3 Ultra 可运行单机装不下的大模型。
Kimi 2.6(1 万亿参数)即使 8-bit 量化也需要多台 Mac 才能加载。分布式推理的驱动力不仅是速度提升,更是模型能否加载到可用内存中——容量问题先于性能问题。
训练时 batch-size 按设备数线性缩放,每台处理相同样本数。以 9B 参数的 Qwen 3.5 微调为例,对比单机和集群的性能差异,展示分布式训练的实际收益。
328 提供单机计算基础,232 提供 agentic 集成,233 提供分布式扩展。三层覆盖 MLX 的完整能力范围——从数组运算到多机集群。
MLX 分布式支持 Thunderbolt 和 Ethernet 两种连接方式。Thunderbolt RDMA 提供低延迟高带宽通信,4 节点加速比达 3 倍。适合私有 AI 集群场景。