楼主:
dnol (舞秋风 忆白云)
2025-07-23 13:11:36大跃进!苹果宣布MLX框架将支援NVIDIA CUDA平台
2025-07-16 17:07:35
记者孟圆琦/编译
https://www.technice.com.tw/issues/ai/182711/
苹果专为Apple Silicon设计的机器学习框架MLX,目前正积极导入NVIDIA CUDA 平台支援
,此举对于机器学习领域而言,无疑是一项重大进展。
MLX 是一个专为在 Apple 芯片上进行高效、灵活的机器学习研究而设计的阵列框架。如
今宣告将导入NVIDIA CUDA 平台支援。
经外媒《AppleInsider》报导指出,这项工作的推动者是 GitHub 上的开发者@zcbenz,
他早在数月前便开始着手 CUDA 支援的原型开发。随后,他将专案拆分成多个较小部分,
并逐步将其整合至 Apple MLX 的主分支中。尽管 CUDA 后端目前仍在开发阶段,但多项
核心运算操作已获得支援并通过测试,包括矩阵乘法、Softmax、归约、排序和索引等。
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 开发人员都乐于听到的消息。
心得/评论:
这则新闻出来有一个星期了,但我没看到版上有人贴。
所以我贴出来,想趁机讨论一下。
我刚去查証了一下,这消息似乎是小道消息,苹果似乎没有正式的发怖这个消息。
若苹果打算将他们内部的AI架构port到cuda上。
这是不是暗示,苹果以后也可能拥抱nvidia生态系。
以前就有听说,苹果很排斥nvidia,因为以前有过节。
难道不久的未来,我们会看到大合解?
nvidia和苹果的股价一起up up?