macOS M2 安装 Jax (jax-metal)

Apple 官方教程:Accelerated JAX on Mac
https://developer.apple.com/metal/jax/


使用 pip install jaxpip install jax-metal 安装 Jax 后,运行验证代码:

python -c 'import jax; print(jax.numpy.arange(10))'

会提示

loc("-":0:0): error: current mps dialect version is 1.0.0, can't parse version 1.1.0 

根据 issue&

你可能感兴趣的:(实践笔记,macos,jax,jax-metal,mac,metal,sonama,mps)