?蘋果版CUDA來了！專為自家芯片打造，M3 Max可跑每秒迭代2.8次

AIGC動態(tài)歡迎閱讀

原標(biāo)題：?蘋果版CUDA來了！專為自家芯片打造，M3 Max可跑每秒迭代2.8次

關(guān)鍵字：機(jī)器,蘋果,框架,模型,研究人員

文章來源：新智元

內(nèi)容字?jǐn)?shù)：4533字

內(nèi)容摘要：

新智元報道編輯：Mindy
【新智元導(dǎo)讀】蘋果新發(fā)布適用于蘋果芯片的機(jī)器學(xué)習(xí)框架MLX，MLX是一個類似于NumPy的陣列框架，旨在為使用蘋果硬件的研究人員簡化 ML 模型的訓(xùn)練和部署。一直以來，英偉達(dá)CUDA稱霸天下，使得它在AI領(lǐng)域的勢力不可動搖。
對于許多人來說，都希望打破這個護(hù)城河，比如PyTorch支持更多GPU，OpenAI的Triton等等。
近日，蘋果針對使用蘋果芯片的機(jī)器學(xué)習(xí)研究人員，專門發(fā)布了一款機(jī)器學(xué)習(xí)框架MLX。
蘋果機(jī)器學(xué)習(xí)團(tuán)隊(duì)的Awni Hannun在X-note中將該軟件稱為：「……專為蘋果芯片設(shè)計(jì)的高效機(jī)器學(xué)習(xí)框架」。
MLX使用起來類似PyTorch、Jax和ArrayFire等現(xiàn)有框架。
但是，MLX針對Apple芯片進(jìn)行了優(yōu)化。以前受困于艱難使用M芯片GPU跑模型的研究人員，現(xiàn)在終于可以擺脫這個苦惱了！
同時，MLX增加了對統(tǒng)一內(nèi)存模型的支持，這也意味著陣列位于共享內(nèi)存中，并且可以在任何支持的設(shè)備類型上執(zhí)行操作，而無需執(zhí)行數(shù)據(jù)復(fù)制。那么，一旦蘋果的其他產(chǎn)品也能夠開始跑模型……有沒有很期待！
英偉達(dá)高級科學(xué)家Jim Fan表示，這可能是蘋果公司迄今

原文鏈接：點(diǎn)此閱讀原文：?蘋果版CUDA來了！專為自家芯片打造，M3 Max可跑每秒迭代2.8次

聯(lián)系作者

文章來源：新智元

作者微信：AI_era

作者簡介：智能+中國主平臺，致力于推動中國從互聯(lián)網(wǎng)+邁向智能+新紀(jì)元。重點(diǎn)關(guān)注人工智能、機(jī)器人等前沿領(lǐng)域發(fā)展，關(guān)注人機(jī)融合、人工智能和機(jī)器人對人類社會與文明進(jìn)化的影響，領(lǐng)航中國新智能時代。