Echo

Echo 关注TA

大家好,我是Echo!

Echo

Echo

关注TA

大家好,我是Echo!

  •  普罗旺斯
  • 自由职业
  • 写了289,631,606字

该文章投稿至Nemo社区   资讯  板块 复制链接


mlx-graphs 项目:Apple Silicon 芯片训练大型数据集速度可提高 10 倍

发布于 2024/02/21 08:04 111浏览 0回复 537

IT之家 2 月 21 日消息,博士生团队近日推出了 mlx-graphs 项目,重点改善 Apple Silicon 芯片上的神经网络处理速度,让其训练大型数据集的速度提高 10 倍。

IT之家从报道中获悉,mlx-graphs 项目由博士生特里斯坦・比洛特(Tristan Bilot)、弗朗切斯科・法里纳(Francesco Farina)和 MLX(专为 Apple Silicon 发布的图形神经网络库)团队共同推进,该开发库的初衷是在 Apple Silicon 芯片上更高效运行让图形神经网络(GNN)。

比洛特表示启用 mlx-graphs 项目之后,Apple Silicon 芯片训练大型图形数据集时,初始基准运行速度可达 PyTorch Geometric 和 DGL 等框架的十倍。

比洛特表示该项目仍有很大的改进空间,从而更好地发挥 Apple Silicon 芯片的能力,相关开发库已托管在 GitHub 上,比洛特欢迎他人探索和测试该库,提交反馈以及 PR(pull requests)。

广告声明:文内含有的对外跳转链接(包括不限于超链接、二维码、口令等形式),用于传递更多信息,节省甄选时间,结果仅供参考,IT之家所有文章均包含本声明。


本文由LinkNemo爬虫[Echo]采集自[https://www.ithome.com/0/751/147.htm]

点赞(0)
点了个评