苹果公司近日开源了其机器学习框架MLX的分布式训练功能,这一举措旨在为AI开发者提供更高效、更灵活的模型训练方案,并挑战NVIDIA在AI计算领域的CUDA生态主导地位。
MLX是苹果在2023年推出的开源机器学习框架,专为Apple Silicon芯片优化。此次开源的分布式训练功能,使得开发者能够利用多台Mac设备或Apple Silicon集群进行大规模模型训练,从而降低对NVIDIA GPU的依赖。苹果表示,MLX的分布式训练支持数据并行和模型并行,并提供了与PyTorch等主流框架的兼容接口,方便开发者迁移现有工作负载。
业界分析认为,苹果此举是其AI战略的重要一步,旨在通过开源生态吸引更多开发者,并推动AI技术在苹果硬件上的应用。尽管目前MLX在生态成熟度上仍无法与CUDA相比,但其在能效和成本上的优势,可能为AI开发提供新的选择。