【ITBEAR科技資訊】2月21日消息，近日，一個(gè)博士生團(tuán)隊(duì)引領(lǐng)了一項(xiàng)名為"mlx-graphs"的創(chuàng)新項(xiàng)目，該項(xiàng)目致力于優(yōu)化Apple Silicon芯片上的神經(jīng)網(wǎng)絡(luò)處理效能。據(jù)稱，通過(guò)這一項(xiàng)目的實(shí)施，Apple Silicon芯片在處理大型數(shù)據(jù)集時(shí)的訓(xùn)練速度有望提升高達(dá)十倍。
該項(xiàng)目由博士生特里斯坦?比洛特(Tristan Bilot)和弗朗切斯科?法里納(Francesco Farina)以及專為Apple Silicon設(shè)計(jì)的圖形神經(jīng)網(wǎng)絡(luò)庫(kù)MLX團(tuán)隊(duì)共同推動(dòng)。他們的初衷是為了讓圖形神經(jīng)網(wǎng)絡(luò)(GNN)在Apple Silicon芯片上運(yùn)行得更高效。
據(jù)ITBEAR科技資訊了解，比洛特在介紹該項(xiàng)目時(shí)表示，啟用mlx-graphs后，Apple Silicon芯片在訓(xùn)練大型圖形數(shù)據(jù)集時(shí)的初始基準(zhǔn)運(yùn)行速度，可以達(dá)到當(dāng)前主流框架如PyTorch Geometric和DGL等的十倍。這一顯著的性能提升預(yù)示著Apple Silicon芯片在神經(jīng)網(wǎng)絡(luò)處理領(lǐng)域的巨大潛力。
然而，比洛特也強(qiáng)調(diào)，該項(xiàng)目仍有很大的改進(jìn)空間，他們期待通過(guò)持續(xù)的優(yōu)化，進(jìn)一步發(fā)揮Apple Silicon芯片的能力。目前，相關(guān)的開發(fā)庫(kù)已經(jīng)托管在GitHub上，他鼓勵(lì)開發(fā)者們積極探索和測(cè)試該庫(kù)，并提供反饋以及pull requests，共同推動(dòng)這一項(xiàng)目的發(fā)展。
