當前位置: 華文世界 > 科技

mlx-graphs 專案:Apple Silicon 訓練大型數據集速度可提高10 倍

2024-02-21科技

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)。