​苹果版CUDA来了！专为自家芯片打造，M3 Max可跑每秒迭代2.8次

AIGC动态欢迎阅读

原标题：​苹果版CUDA来了！专为自家芯片打造，M3 Max可跑每秒迭代2.8次

关键字：机器,苹果,框架,模型,研究人员

文章来源：新智元

内容字数：4533字

内容摘要：

新智元报道编辑：Mindy
【新智元导读】苹果新发布适用于苹果芯片的机器学习框架MLX，MLX是一个类似于NumPy的阵列框架，旨在为使用苹果硬件的研究人员简化 ML 模型的训练和部署。一直以来，英伟达CUDA称霸天下，使得它在AI领域的势力不可动摇。
对于许多人来说，都希望打破这个护城河，比如PyTorch支持更多GPU，OpenAI的Triton等等。
近日，苹果针对使用苹果芯片的机器学习研究人员，专门发布了一款机器学习框架MLX。
苹果机器学习团队的Awni Hannun在X-note中将该软件称为：「……专为苹果芯片设计的高效机器学习框架」。
MLX使用起来类似PyTorch、Jax和ArrayFire等现有框架。
但是，MLX针对Apple芯片进行了优化。以前受困于艰难使用M芯片GPU跑模型的研究人员，现在终于可以摆脱这个苦恼了！
同时，MLX增加了对统一内存模型的支持，这也意味着阵列位于共享内存中，并且可以在任何支持的设备类型上执行操作，而无需执行数据复制。那么，一旦苹果的其他产品也能够开始跑模型……有没有很期待！
英伟达高级科学家Jim Fan表示，这可能是苹果公司迄今

原文链接：点此阅读原文：​苹果版CUDA来了！专为自家芯片打造，M3 Max可跑每秒迭代2.8次

联系作者

文章来源：新智元

作者微信：AI_era

作者简介：智能+中国主平台，致力于推动中国从互联网+迈向智能+新纪元。重点关注人工智能、机器人等前沿领域发展，关注人机融合、人工智能和机器人革命对人类社会与文明进化的影响，领航中国新智能时代。