SmolCluster:基于 Socket 的跨异构硬件分布式训练库

retr0jirachi · x · 2026-09-01

SmolCluster 是一个用于异构硬件的分布式深度学习库,旨在帮助开发者从零开始学习分布式训练和推理。该项目仅使用 Python 的 socket 库,无需 NCCL 或 MPI,可在 Mac mini、树莓派、MacBook 和 Windows 机器上运行。支持 FSDP、ClassicDP、EDP、模型并行等多种分布式算法。作者强调这是一个教育性质的项目,代码优先考虑可读性而非性能。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →