MLX:苹果 专为统一内存架构(UMA) 设计的机器学习框架

2023-12-16 23:36:52

“晨兴理荒秽,带月荷锄归”

夜深闻讯,有点兴奋~

苹果为 UMA 设计的深度学习框架真的来了
统一内存架构 得益于 CPU 与 GPU 内存的共享,同时与 MacOS 和 M 芯片 交相辉映,在效率上,实现对其他框架的降维打击。?

搓手学习,有点精彩👐🏻

60d42a29fd6a45d88f32bf389c3af747.png

MLX 地址

翻译:

MLX是苹果硅片上机器学习的数组框架,由苹果机器学习研究团队带来。

MLX的一些关键特性包括:

熟悉的API:MLX具有与NumPy非常相似的Python API。MLX还有一个功能齐全的C++ API,与Python API非常相似。MLX拥有像mlx.nn和mlx.optimizers这样的高级包,其API紧密跟随PyTorch,以简化构建更复杂的模型。

可组合函数转换:MLX支持自动微分、自动向量化和计算图优化的可组合函数转换。

延迟计算:MLX中的计算是延迟的。数组只有在需要时才实现。

动态图构建:MLX中的计算图是动态构建的。更改函数参数的形状不会触发慢速编译,调试简单直观。

多设备:操作可以在任何支持的设备上运行(目前是CPU和GPU)。

统一内存:MLX与其他框架的一个显著区别是统一内存模型。MLX中的数组存在于共享内存中。在任何支持的设备类型上对MLX数组进行的操作都无需传输数据。

MLX由机器学习研究人员为机器学习研究人员设计。该框架旨在用户友好,但仍然高效地训练和部署模型。框架本身的设计也概念上很简单。我们的目标是让研究人员能够轻松扩展和改进MLX,以快速探索新想法。

MLX的设计灵感来自于像NumPy、PyTorch、Jax和ArrayFire这样的框架。

文章来源:https://blog.csdn.net/Gloamirror/article/details/135038859
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。