rocm-jaxlib-v0.4.28
·
5653 commits
to main
since this release
ROCm jax/jaxlib 0.4.28 Release
ROCm 6.1.1
jaxlib
Python 3.9
python3 -m pip install https://github.com/ROCm/jax/releases/download/rocm-jaxlib-v0.4.28/jaxlib-0.4.28+rocm611-cp39-cp39-manylinux2014_x86_64.whl
Python 3.10
python3 -m pip install https://github.com/ROCm/jax/releases/download/rocm-jaxlib-v0.4.28/jaxlib-0.4.28+rocm611-cp310-cp310-manylinux2014_x86_64.whl
Python 3.11
python3 -m pip install https://github.com/ROCm/jax/releases/download/rocm-jaxlib-v0.4.28/jaxlib-0.4.28+rocm611-cp311-cp311-manylinux2014_x86_64.whl
JAX
python3 -m pip install https://github.com/ROCm/jax/archive/refs/tags/rocm-jaxlib-v0.4.28.tar.gz
Docker Images
Python 3.9
docker pull rocm/jax:rocm6.1.1-jax0.4.28-py3.9.19
Python 3.10
docker pull rocm/jax:rocm6.1.1-jax0.4.28-py3.10.14
Python 3.11
docker pull rocm/jax:rocm6.1.1-jax0.4.28-py3.11.9
ROCm 6.0.2
jaxlib
Python 3.9
python3 -m pip install https://github.com/ROCm/jax/releases/download/rocm-jaxlib-v0.4.28/jaxlib-0.4.28+rocm602-cp39-cp39-manylinux2014_x86_64.whl
Python 3.10
python3 -m pip install https://github.com/ROCm/jax/releases/download/rocm-jaxlib-v0.4.28/jaxlib-0.4.28+rocm602-cp310-cp310-manylinux2014_x86_64.whl
Python 3.11
python3 -m pip install https://github.com/ROCm/jax/releases/download/rocm-jaxlib-v0.4.28/jaxlib-0.4.28+rocm602-cp311-cp311-manylinux2014_x86_64.whl
JAX
python3 -m pip install https://github.com/ROCm/jax/archive/refs/tags/rocm-jaxlib-v0.4.28.tar.gz
Docker Images
Python 3.9
docker pull rocm/jax:rocm6.0.2-jax0.4.28-py3.9.19
Python 3.10
docker pull rocm/jax:rocm6.0.2-jax0.4.28-py3.10.14
Python 3.11
docker pull rocm/jax:rocm6.0.2-jax0.4.28-py3.11.9