AirTrain

AirTrain 是一个专为 Apple Silicon Mac 设备设计的分布式机器学习训练框架，支持在多台本地 Mac 上协同训练大模型。其核心采用 DiLoCo（Distributed Low-Communication）算法，将梯度同步频率降低至每 500 步一次，并仅传输参数差值，使网络通信量减少 500 倍，从而实现在 Wi-Fi 环境下的高效分布式训练；集成零配置 mDNS 发现、容错节点动态加入/退出、检查点中继与本地 Web 仪表盘等功能。适用于教育实验、低成本模型微调、边缘协作训练及资源受限场景下的本地化 ML 开发。