AIGC動態歡迎閱讀
原標題:?蘋果版CUDA來了!專為自家芯片打造,M3 Max可跑每秒迭代2.8次
文章來源:新智元
內容字數:4533字
內容摘要:
新智元報道編輯:Mindy
【新智元導讀】蘋果新發布適用于蘋果芯片的機器學習框架MLX,MLX是一個類似于NumPy的陣列框架,旨在為使用蘋果硬件的研究人員簡化 ML 模型的訓練和部署。一直以來,英偉達CUDA稱霸天下,使得它在AI領域的勢力不可動搖。
對于許多人來說,都希望打破這個護城河,比如PyTorch支持更多GPU,OpenAI的Triton等等。
近日,蘋果針對使用蘋果芯片的機器學習研究人員,專門發布了一款機器學習框架MLX。
蘋果機器學習團隊的Awni Hannun在X-note中將該軟件稱為:「……專為蘋果芯片設計的高效機器學習框架」。
MLX使用起來類似PyTorch、Jax和ArrayFire等現有框架。
但是,MLX針對Apple芯片進行了優化。以前受困于艱難使用M芯片GPU跑模型的研究人員,現在終于可以擺脫這個苦惱了!
同時,MLX增加了對統一內存模型的支持,這也意味著陣列位于共享內存中,并且可以在任何支持的設備類型上執行操作,而無需執行數據復制。那么,一旦蘋果的其他產品也能夠開始跑模型……有沒有很期待!
英偉達高級科學家Jim Fan表示,這可能是蘋果公司迄今
原文鏈接:點此閱讀原文:?蘋果版CUDA來了!專為自家芯片打造,M3 Max可跑每秒迭代2.8次
聯系作者
文章來源:新智元
作者微信:AI_era
作者簡介:智能+中國主平臺,致力于推動中國從互聯網+邁向智能+新紀元。重點關注人工智能、機器人等前沿領域發展,關注人機融合、人工智能和機器人對人類社會與文明進化的影響,領航中國新智能時代。
? 版權聲明
文章版權歸作者所有,未經允許請勿轉載。
相關文章
暫無評論...