mlx-graphs項目:Apple Silicon芯片訓練大型數據集速度可提高10倍
2024-02-21 08:27:02 IT之家2 月 21 日消息,博士生團隊近日推出了 mlx-graphs 項目,重點改善 Apple Silicon 芯片上的神經網絡處理速度,讓其訓練大型數據集的速度提高 10 倍。
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)。
免責聲明:本文由作者原創。文章內容系作者個人觀點,轉載目的在于傳遞更多信息,并不代表EETOP贊同其觀點和對其真實性負責。如涉及作品內容、版權和其它問題,請及時聯系我們,我們將在第一時間刪除!