昨天,苹果的机器学习研究团队发布了一个新的开源模型运行软件框架,名为 MLX。这是一个专为苹果 M 系列芯片设计的高效机器学习框架,旨在让配备 M 系列芯片的设备能够更有效地进行深度学习训练。
据 GitHub 主页显示,MLX 包括 Python API 和 C++ API,具有用于自动微分、自动矢量化和计算图优化的可组合函数转换。这个框架采用惰性计算,动态构建计算图,支持多设备,并且统一内存。
英伟达高级科学家 Jim Fan 指出,这可能是苹果迄今为止在开源人工智能方面的最大举措。此举显示出苹果在开源领域的承诺和投入,进一步推动了人工智能的发展。