📱 Apple MLX 框架引入 NVIDIA CUDA 支持

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

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

9to5Mac

💢 Only 🖥 Can Do

🍀频道 🍵茶馆 📮投稿
来自频道: @zaihuapd
Loading comments...