Appleinsider 發(fā)現(xiàn),蘋果專為 Apple Silicon 設(shè)計的機(jī)器學(xué)習(xí)框架 MLX 正新增 CUDA 支持。
這意味著開發(fā)者可直接在 Apple Silicon Mac 上利用 MLX 開發(fā)應(yīng)用,隨后導(dǎo)出至 CUDA 環(huán)境運(yùn)行。
在此之前,由于 MLX 深度集成于蘋果平臺的 Metal 框架,其無法在 macOS 之外運(yùn)行,故導(dǎo)致開發(fā)者需購買額外硬件以完成部署和測試。
IT之家注意到,該項目由 GitHub 開發(fā)者 @zcbenz 主導(dǎo),其數(shù)月前開始構(gòu)建 CUDA 相關(guān)支持代碼,隨后將項目拆分為多個模塊,逐步并入蘋果 MLX 主分支。
需明確說明的是,此支持僅限從適配蘋果環(huán)境的 MLX 框架導(dǎo)出代碼,使其可在 Nvidia 顯卡及服務(wù)器硬件運(yùn)行。這既不意味著 Mac Pro 或外接顯卡塢可連接 Nvidia 顯卡本地運(yùn)行機(jī)器學(xué)習(xí)應(yīng)用,也無法讓基于 CUDA 編寫的項目直接在 Apple Silicon 上運(yùn)行。
對開發(fā)者而言,此項目優(yōu)勢在于成本控制 ——Nvidia 硬件配置成本極高,甚至數(shù)倍于頂配 Mac 售價。對于小規(guī)模開發(fā)者,組織可通過 Apple Silicon 設(shè)備進(jìn)行開發(fā)測試,僅在量產(chǎn)階段使用 Nvidia 硬件以節(jié)省開支。
此外,MLX 項目在 CUDA 系統(tǒng)運(yùn)行時性能遠(yuǎn)超 Mac,結(jié)合 Nvidia 硬件在機(jī)器學(xué)習(xí)領(lǐng)域的廣泛應(yīng)用,開發(fā)者有望獲得更強(qiáng)算力支持。