jax与jaxlib与Mac arm架构不符，Mac如何下载这些

JAX是一个基于NumPy的高性能张量库，而jaxlib则是它的底层硬件加速库，通常适用于CUDA和TPU。然而，由于Apple的M1芯片（ARM架构）不直接支持CUDA，你需要找到兼容该架构的版本。以下是解决此问题的一般步骤：

1. **JAX for M1**: JAX本身已提供针对macOS M1的支持。你可以直接在M1 Mac上使用pip安装最新版的JAX，无需特别处理：

   pip install jax

检查是否安装成功可以用 `python -m jax` 或 `python -c 'import jax; print(jax.__version__)'`。

2. **jaxlib**: 对于jaxlib，目前主要支持CUDA的版本。如果你希望利用GPU加速，你可以在M1上通过Docker或其他虚拟化工具如Vagrant或Wine来运行有CUDA支持的操作系统（比如Linux或Windows Subsystem for Linux，WSL）并安装jupyter notebook或Google Colab，然后在那里安装jaxlib。

3. **替代方案**：如果不需要GPU加速，那么可以直接使用CPU版本的JAX和无依赖的jaxlib。你也可以尝试寻找是否有专门为arm架构编译的第三方包，但这可能会比较少见。

4. **GitHub问题追踪**: 如果你在安装过程中遇到特定问题，可以在JAX或jaxlib的GitHub仓库（https://github.com/google/jax 和 https://github.com/google/jaxlib）的 issues 页面查找解决方案或报告新的问题。