Skip to main content

Apple MLX 框架引入 NVIDIA CUDA 支持Apple 的 MLX 机器学习框架正在添加对 NVIDIA CUDA 平台的支持，开发者可直接在配备 NVIDIA GPU 的系统上运行 MLX 模型

  1. Apple MLX 框架引入 NVIDIA CUDA 支持

    Apple 的 MLX 机器学习框架正在添加对 NVIDIA CUDA 平台的支持，开发者可直接在配备 NVIDIA GPU 的系统上运行 MLX 模型。目前已实现矩阵乘法、softmax 等核心运算，相关代码已逐步合并进主分支。

    未来研究人员和工程师可在 Mac 上原型开发，并便捷地将模型部署至主流的 NVIDIA GPU 集群。尽管所有运算符尚未完全实现，AMD GPU 支持也尚未上线，但此次升级将加速测试与研究流程。

    9to5Mac

    Only Can Do

    🍀频道 🍵茶馆 📮投稿
    ❤️ 135 👍 48 😁 16 🤯 12 👎 5 💊 5