专为Apple Silicon优化的开源机器学习框架:MLX (Machine Learning eXtension)

  • MLX是一个专为Apple Silicon优化的开源机器学习框架,由苹果的机器学习研究团队开发。它旨在提供简单友好的使用方法,并充分利用Apple Silicon芯片(如M1、M2、M3、M4等)的硬件性能。
  • MLX支持NumPy风格的数组操作和PyTorch风格的高层次API,提供了自动微分、自动向量化和计算图优化等功能。它可以在Apple Silicon的CPU和GPU上运行,并通过统一内存模型避免数据在CPU和GPU间的重复拷贝。

mac下安装mlx

需要条件:

使用M系列芯片(Apple Silicon),原生支持Python >= 3.9,macOS >= 13.5

安装手册:

Build and Install — MLX 0.23.0 documentation

直接pip安装

pip install mlx

但是如果mac是Intel芯片,好像就没法装了。

查询自己的mac是哪种芯片:

python -c "import platform; print(platform.processor())"

会输出arm或者i386 

也可以使用uname查询

uname -p

MLX能够充分利用Apple Silicon芯片的高性能和能效比,提供更快的训练和推理速度。比如在安装exo的时候,可能就需要安装mlx库。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值