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

---

使用 pip install jax 或 pip 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&