蘋果公司近期在機(jī)器學(xué)習(xí)領(lǐng)域邁出了重要一步，其專為Apple Silicon設(shè)計的機(jī)器學(xué)習(xí)框架MLX，現(xiàn)已加入對英偉達(dá)CUDA的支持。這一變化意味著，開發(fā)者能夠利用MLX在Apple Silicon Mac上進(jìn)行應(yīng)用開發(fā)，隨后將應(yīng)用無縫遷移至CUDA環(huán)境運(yùn)行。

以往，由于MLX框架與蘋果的metal平臺深度綁定，開發(fā)者在macOS系統(tǒng)之外難以進(jìn)行相關(guān)工作，往往需要額外購置硬件以滿足部署和測試需求。而現(xiàn)在，CUDA支持的加入打破了這一局限，開發(fā)者可以在蘋果設(shè)備上進(jìn)行開發(fā)和初步測試，再轉(zhuǎn)向高性能的CUDA硬件進(jìn)行實(shí)際應(yīng)用部署。

此次更新得益于GitHub開發(fā)者@zcbenz的努力，他數(shù)月前開始著手CUDA支持代碼的開發(fā)，并經(jīng)過一系列模塊拆分后，成功將其并入蘋果的MLX主分支。但值得注意的是，這一支持僅限于從適配蘋果環(huán)境的MLX框架導(dǎo)出代碼，在配備Nvidia顯卡的服務(wù)器硬件上運(yùn)行。也就是說，Mac Pro或外接顯卡塢并不支持直接連接Nvidia顯卡以本地運(yùn)行機(jī)器學(xué)習(xí)應(yīng)用，開發(fā)者仍需在完成開發(fā)后進(jìn)行代碼的適配與導(dǎo)出。

對于開發(fā)者而言，這一新功能具有重大意義。Nvidia硬件配置成本高昂，往往遠(yuǎn)超頂配Mac的價格。因此，小型開發(fā)團(tuán)隊(duì)可以在Apple Silicon設(shè)備上進(jìn)行開發(fā)和測試，僅在量產(chǎn)階段才需要借助Nvidia硬件，從而有效控制成本。在CUDA環(huán)境下，MLX項(xiàng)目的運(yùn)行性能通常遠(yuǎn)超Mac，結(jié)合Nvidia硬件在機(jī)器學(xué)習(xí)領(lǐng)域的廣泛應(yīng)用，開發(fā)者將有機(jī)會獲得更強(qiáng)大的算力支持，進(jìn)一步提升應(yīng)用的性能和效果。

蘋果MLX框架新增CUDA支持，不僅為開發(fā)者提供了更多便利和靈活性，也降低了機(jī)器學(xué)習(xí)應(yīng)用開發(fā)的成本。這一舉措預(yù)計將吸引更多開發(fā)者的關(guān)注和參與，推動機(jī)器學(xué)習(xí)領(lǐng)域的發(fā)展。