大躍進！蘋果宣布MLX框架將支援NVIDIA CUDA平台

記者孟圓琦／編譯

蘋果專為Apple Silicon設計的機器學習框架MLX，目前正積極導入NVIDIA CUDA 平台支援，此舉對於機器學習領域而言，無疑是一項重大進展。

經外媒《AppleInsider》報導指出，這項工作的推動者是 GitHub 上的開發者@zcbenz，他早在數月前便開始著手 CUDA 支援的原型開發。隨後，他將專案拆分成多個較小部分，並逐步將其整合至 Apple MLX 的主分支中。儘管 CUDA 後端目前仍在開發階段，但多項核心運算操作已獲得支援並通過測試，包括矩陣乘法、Softmax、歸約、排序和索引等。

延伸閱讀：
蘋果加入AI戰　推出自家晶片設計的MLX機器學習框架
告別卡頓！蘋果自研C1晶片2大關鍵優勢 重啟手機通訊新篇章
蘋果MacBook Pro M5再延期！更新大改款等2026見

CUDA (Compute Unified Device Architecture)，本質上是 NVIDIA 硬體的「Metal」：一個專為自家 GPU 設計的運算平台，旨在最大化其在高效率平行運算任務中的效能。對於許多人而言，CUDA 是在 NVIDIA GPU 上執行機器學習工作負載的標準方式，並廣泛應用於整個機器學習生態系統，從學術研究到商業部署。諸如PyTorch和TensorFlow等深度學習框架，都依賴 CUDA 來利用 GPU 加速運算。

那麼，為何 Apple 的 MLX 框架要開始支援 CUDA 呢？MLX 最初是為 Apple Silicon 和 Metal 進行優化，但增加 CUDA 後端改變了這一局面。現在，研究人員和工程師可以在 Mac 上利用 MLX 本地開發 CUDA 模型，然後將其部署到大型 NVIDIA GPU 叢集上，這些叢集目前在機器學習訓練工作負載方面仍佔據主導地位。

儘管如此，目前仍存在一些限制，其中大部分仍在積極開發中。例如，並非所有 MLX 運算子都已實作完成，且對 AMD GPU 的支援仍在更遠的計畫中。

即便如此，能夠在無需重寫程式碼的情況下，將 MLX 程式碼引入 NVIDIA GPU，為更快的測試、實驗和研究用例開啟了大門，這幾乎是所有 AI 開發人員都樂於聽到的消息。

資料來源：9to5mac、appleinsider

瀏覽 186 次