蘋果專為Apple Silicon設計的機器學習框架MLX,目前正積極導入NVIDIA CUDA 平台支援
,此舉對於機器學習領域而言,無疑是一項重大進展。
MLX 是一個專為在 Apple 晶片上進行高效、靈活的機器學習研究而設計的陣列框架。如
今宣告將導入NVIDIA CUDA 平台支援。
經外媒《AppleInsider》報導指出,這項工作的推動者是 GitHub 上的開發者@zcbenz,
他早在數月前便開始著手 CUDA 支援的原型開發。隨後,他將專案拆分成多個較小部分,
並逐步將其整合至 Apple MLX 的主分支中。儘管 CUDA 後端目前仍在開發階段,但多項
核心運算操作已獲得支援並通過測試,包括矩陣乘法、Softmax、歸約、排序和索引等。
那麼,為何 Apple 的 MLX 框架要開始支援 CUDA 呢?MLX 最初是為 Apple Silicon 和
Metal 進行優化,但增加 CUDA 後端改變了這一局面。現在,研究人員和工程師可以在
Mac 上利用 MLX 本地開發 CUDA 模型,然後將其部署到大型 NVIDIA GPU 叢集上,這些
叢集目前在機器學習訓練工作負載方面仍佔據主導地位。