@AIGC开发者消息:12月6日,苹果公司悄然发布了一款名为MLX的深度学习框架,MLX 框架专为 Apple 芯片而设计,由 Apple 的机器学习研究团队研发。现在,它已经在GitHub开源,有望成为在 Apple 设备上探索机器学习的开发者的首选框架。
🚀 MLX的GitHub地址:https://github.com/ml-explore/mlx
MLX 的独特之处在于:
🧠 熟悉的 API:MLX 的 API 设计与 NumPy 和 PyTorch 相似,使得用户可以方便地构建和训练复杂的机器学习模型,更容易上手。
⏱ 高效延迟计算:MLX 中的计算是延迟进行的,数组只在必要时才实体化,避免了不必要的计算,提升了资源效率。
🔁 动态图构建:MLX 的计算图是动态构建的,适应输入形状的变化,使调试和实验变得简单直观。
💻 多设备支持:MLX 可以无缝地利用 Apple 设备的 CPU 和 GPU,让您的硬件性能发挥到极致。
💾 统一内存模型:与其他框架不同,MLX 在共享内存中存储数组,消除了设备间数据的移动,进一步加速了操作流程。
MLX 如何增强 AI 能力?
📝 MLX 展示了其在高效训练 Transformer 模型和增强自然语言处理方面的强大能力。MLX 示例存储库提供了一些示例,包括:
- Transformer 语言模型训练
- 使用Mistral 生成文本
- 通过 LoRA 进行参数微调
- 使用Stable Diffusion技术的图像生成
- 使用 OpenAI 的 Whisper 进行语音识别
MLX 框架可以在 Apple 产品上通过 pip 安装而无需其他依赖包即可原生运行。随着 MLX 的推出,它有望成为推动 Apple 设备上机器学习研究的首选框架。