Apple Silicon通过mlx-graphs项目大幅提升人工智能训练速度

一个旨在提高AppleSilicon上神经网络处理速度的新项目有可能将大型数据集的训练速度提高十倍。创建机器学习项目的问题之一是在大型数据集上训练模型。这依赖于大量的计算能力来啃噬数据，但这方面的改进有助于加快训练速度，并有可能改进模型。

mlx-graphs是博士生特里斯坦-比洛特（TristanBilot）、弗朗切斯科-法里纳（FrancescoFarina）和MLX团队的一个新项目，是一个旨在帮助图形神经网络（GNN）在AppleSilicon上更高效运行的库。图形神经网络用于预测节点和边，并执行基于图形的任务，在计算机视觉领域尤其有用。

mlx-graphs项目以MLX为基础，是专为AppleSilicon发布的图形神经网络库。该项目旨在为该领域的研究人员提供可观的性能提升。

Bilot声称，在对大型图形数据集进行训练时，该库的初始基准运行速度可达PyTorchGeometric和DGL等框架的十倍。它通过使用专门设计的内核来并行化直接在M系列芯片GPU上运行的GNN计算。

苹果公司在MLX图形方面的工作仍处于初期阶段

Bilot承认，"仍有很大的空间可以做出重大贡献"。这可能是一个暗示，即随着进一步的开发，可能会发现更多的速度提升。

mlx-graphs库可在GitHub上下载和安装：

https://github.com/mlx-graphs/mlx-graphs

Bilot邀请其他人探索和测试该库，提供反馈，并通过拉取请求提交实现。该项目是对机器学习和生成式人工智能兴趣浪潮的一部分，这一领域可以极大地改变内容创作和向用户提供信息的方式。

就苹果公司而言，公司内部的研究人员已经创建了一个用于动画图像的生成式人工智能工具。此外，其他项目也在Xcode工具中测试人工智能的使用。

苹果公司首席执行官蒂姆-库克（TimCook）也谈到，苹果公司将在2024年晚些时候向用户推出大量人工智能功能。