Releases: ROCm/jax
rocm-jaxlib-v0.4.30
ROCm jax/jaxlib 0.4.30 Release
ROCm 6.1.1
jaxlib
Python 3.9
python3 -m pip install https://github.com/ROCm/jax/releases/download/rocm-jaxlib-v0.4.30/jaxlib-0.4.30+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.30/jaxlib-0.4.30+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.30/jaxlib-0.4.30+rocm611-cp311-cp311-manylinux2014_x86_64.whl
JAX
python3 -m pip install https://github.com/ROCm/jax/archive/refs/tags/rocm-jaxlib-v0.4.30.tar.gz
Docker Images
Python 3.9
docker pull rocm/jax-community:rocm6.1.1-jax0.4.30-py3.9.19
Python 3.10
docker pull rocm/jax-community:rocm6.1.1-jax0.4.30-py3.10.14
Python 3.11
docker pull rocm/jax-community:rocm6.1.1-jax0.4.30-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.30/jaxlib-0.4.30+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.30/jaxlib-0.4.30+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.30/jaxlib-0.4.30+rocm602-cp311-cp311-manylinux2014_x86_64.whl
JAX
python3 -m pip install https://github.com/ROCm/jax/archive/refs/tags/rocm-jaxlib-v0.4.30.tar.gz
Docker Images
Python 3.9
docker pull rocm/jax-community:rocm6.0.2-jax0.4.30-py3.9.19
Python 3.10
docker pull rocm/jax-community:rocm6.0.2-jax0.4.30-py3.10.14
Python 3.11
docker pull rocm/jax-community:rocm6.0.2-jax0.4.30-py3.11.9
rocm-jaxlib-v0.4.29
ROCm jax/jaxlib 0.4.29 Release
ROCm 6.1.1
jaxlib
Python 3.9
python3 -m pip install https://github.com/ROCm/jax/releases/download/rocm-jaxlib-v0.4.29/jaxlib-0.4.29+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.29/jaxlib-0.4.29+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.29/jaxlib-0.4.29+rocm611-cp311-cp311-manylinux2014_x86_64.whl
JAX
python3 -m pip install https://github.com/ROCm/jax/archive/refs/tags/rocm-jaxlib-v0.4.29.tar.gz
Docker Images
Python 3.9
docker pull rocm/jax-community:rocm6.1.1-jax0.4.29-py3.9.19
Python 3.10
docker pull rocm/jax-community:rocm6.1.1-jax0.4.29-py3.10.14
Python 3.11
docker pull rocm/jax-community:rocm6.1.1-jax0.4.29-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.29/jaxlib-0.4.29+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.29/jaxlib-0.4.29+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.29/jaxlib-0.4.29+rocm602-cp311-cp311-manylinux2014_x86_64.whl
JAX
python3 -m pip install https://github.com/ROCm/jax/archive/refs/tags/rocm-jaxlib-v0.4.29.tar.gz
Docker Images
Python 3.9
docker pull rocm/jax-community:rocm6.0.2-jax0.4.29-py3.9.19
Python 3.10
docker pull rocm/jax-community:rocm6.0.2-jax0.4.29-py3.10.14
Python 3.11
docker pull rocm/jax-community:rocm6.0.2-jax0.4.29-py3.11.9
rocm-jaxlib-v0.4.28
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
Jax/Jaxlib 0.4.26 release for ROCm
**Jax/Jaxlib 0.4.26 release for ROCm
Note: Added ROCm 6.1 version on May 6th, 2024
ROCm 6.1
Install Jaxlib
Python 3.9
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.26/jaxlib-0.4.26+rocm610-cp39-cp39-manylinux2014_x86_64.whl
Python 3.10
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.26/jaxlib-0.4.26+rocm610-cp310-cp310-manylinux2014_x86_64.whl
Python 3.11
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.26/jaxlib-0.4.26+rocm610-cp311-cp311-manylinux2014_x86_64.whl
Install Jax
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/archive/refs/tags/jaxlib-v0.4.26.tar.gz
Docker Images
Python 3.9
docker pull rocm/jax:rocm6.1.0-jax0.4.26-py3.9.0
Python 3.10
docker pull rocm/jax:rocm6.1.0-jax0.4.26-py3.10.0
Python 3.11
docker pull rocm/jax:rocm6.1.0-jax0.4.26-py3.11.0
ROCm 6.0.0 and 5.7.0
Install Jaxlib
Python 3.9
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.26/jaxlib-0.4.26+rocm570-cp39-cp39-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.26/jaxlib-0.4.26+rocm600-cp39-cp39-manylinux2014_x86_64.whl
Python 3.10
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.26/jaxlib-0.4.26+rocm570-cp310-cp310-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.26/jaxlib-0.4.26+rocm600-cp310-cp310-manylinux2014_x86_64.whl
Python 3.11
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.26/jaxlib-0.4.26+rocm570-cp311-cp311-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.26/jaxlib-0.4.26+rocm600-cp311-cp311-manylinux2014_x86_64.whl
Install Jax
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/archive/refs/tags/jaxlib-v0.4.26.tar.gz
Docker Images
Python 3.9
docker pull rocm/jax:rocm5.7.0-jax0.4.26-py3.9.0
docker pull rocm/jax:rocm6.0.0-jax0.4.26-py3.9.0
Python 3.10
docker pull rocm/jax:rocm5.7.0-jax0.4.26-py3.10.0
docker pull rocm/jax:rocm6.0.0-jax0.4.26-py3.10.0
Python 3.11
docker pull rocm/jax:rocm5.7.0-jax0.4.26-py3.11.0
docker pull rocm/jax:rocm6.0.0-jax0.4.26-py3.11.0
Jax/Jaxlib 0.4.24 release for ROCm
**Jax/Jaxlib 0.4.24 release for ROCm
Install Jaxlib
Python 3.9
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.24/jaxlib-0.4.24+rocm570-cp39-cp39-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.24/jaxlib-0.4.24+rocm600-cp39-cp39-manylinux2014_x86_64.whl
Python 3.10
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.24/jaxlib-0.4.24+rocm570-cp310-cp310-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.24/jaxlib-0.4.24+rocm600-cp310-cp310-manylinux2014_x86_64.whl
Python 3.11
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.24/jaxlib-0.4.24+rocm570-cp311-cp311-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.24/jaxlib-0.4.24+rocm600-cp311-cp311-manylinux2014_x86_64.whl
Install Jax
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/archive/refs/tags/jaxlib-v0.4.24.tar.gz
Docker Images
Python 3.9
docker pull rocm/jax:rocm5.7.0-jax0.4.24-py3.9.0
docker pull rocm/jax:rocm6.0.0-jax0.4.24-py3.9.0
Python 3.10
docker pull rocm/jax:rocm5.7.0-jax0.4.24-py3.10.0
docker pull rocm/jax:rocm6.0.0-jax0.4.24-py3.10.0
Python 3.11
docker pull rocm/jax:rocm5.7.0-jax0.4.24-py3.11.0
docker pull rocm/jax:rocm6.0.0-jax0.4.24-py3.11.0
Jax/Jaxlib 0.4.25 release for ROCm
**Jax/Jaxlib 0.4.25 release for ROCm
Install Jaxlib
Python 3.9
python3 -m pip install https://github.com/ROCm/jax/releases/download/jaxlib-v0.4.25/jaxlib-0.4.25+rocm600-cp39-cp39-manylinux2014_x86_64.whl
Python 3.10
python3 -m pip install https://github.com/ROCm/jax/releases/download/jaxlib-v0.4.25/jaxlib-0.4.25+rocm600-cp310-cp310-manylinux2014_x86_64.whl
Python 3.11
python3 -m pip install https://github.com/ROCm/jax/releases/download/jaxlib-v0.4.25/jaxlib-0.4.25+rocm600-cp311-cp311-manylinux2014_x86_64.whl
Install Jax
python3 -m pip install https://github.com/ROCm/jax/archive/refs/tags/jaxlib-v0.4.25.tar.gz
Docker Images
Python 3.9
docker pull rocm/jax:rocm6.0.0-jax0.4.25-py3.9.0
Python 3.10
docker pull rocm/jax:rocm6.0.0-jax0.4.25-py3.10.0
Python 3.11
docker pull rocm/jax:rocm6.0.0-jax0.4.25-py3.11.0
Jax/Jaxlib 0.4.23 release for ROCm
**Jax/Jaxlib 0.4.23 release for ROCm
Install Jaxlib
Python 3.9
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.23/jaxlib-0.4.23+rocm570-cp39-cp39-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.23/jaxlib-0.4.23+rocm600-cp39-cp39-manylinux2014_x86_64.whl
Python 3.10
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.23/jaxlib-0.4.23+rocm570-cp310-cp310-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.23/jaxlib-0.4.23+rocm600-cp310-cp310-manylinux2014_x86_64.whl
Python 3.11
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.23/jaxlib-0.4.23+rocm570-cp311-cp311-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.23/jaxlib-0.4.23+rocm600-cp311-cp311-manylinux2014_x86_64.whl
Install Jax
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/archive/refs/tags/jaxlib-v0.4.23.tar.gz
Docker Images
Python 3.9
docker pull rocm/jax:rocm5.7.0-jax0.4.23-py3.9.0
docker pull rocm/jax:rocm6.0.0-jax0.4.23-py3.9.0
Python 3.10
docker pull rocm/jax:rocm5.7.0-jax0.4.23-py3.10.0
docker pull rocm/jax:rocm6.0.0-jax0.4.23-py3.10.0
Python 3.11
docker pull rocm/jax:rocm5.7.0-jax0.4.23-py3.11.0
docker pull rocm/jax:rocm6.0.0-jax0.4.23-py3.11.0
Jax/Jaxlib 0.4.21 release for ROCm
**Jax/Jaxlib 0.4.21 release for ROCm
Install Jaxlib
Python 3.9
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.21/jaxlib-0.4.21+rocm560-cp39-cp39-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.21/jaxlib-0.4.21+rocm570-cp39-cp39-manylinux2014_x86_64.whl
Python 3.10
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.21/jaxlib-0.4.21+rocm560-cp310-cp310-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.21/jaxlib-0.4.21+rocm570-cp310-cp310-manylinux2014_x86_64.whl
Python 3.11
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.21/jaxlib-0.4.21+rocm560-cp311-cp311-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.21/jaxlib-0.4.21+rocm570-cp311-cp311-manylinux2014_x86_64.whl
Install Jax
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/archive/refs/tags/jaxlib-v0.4.21.tar.gz
Docker Images
Python 3.9
docker pull rocm/jax:rocm5.6.0-jax0.4.21-py3.9.0
docker pull rocm/jax:rocm5.7.0-jax0.4.21-py3.9.0
Python 3.10
docker pull rocm/jax:rocm5.6.0-jax0.4.21-py3.10.0
docker pull rocm/jax:rocm5.7.0-jax0.4.21-py3.10.0
Python 3.11
docker pull rocm/jax:rocm5.6.0-jax0.4.21-py3.11.0
docker pull rocm/jax:rocm5.7.0-jax0.4.21-py3.11.0
Jax/Jaxlib 0.4.20 release for ROCm
**Jax/Jaxlib 0.4.20 release for ROCm
Install Jaxlib
Python 3.9
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.20/jaxlib-0.4.20+rocm560-cp39-cp39-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.20/jaxlib-0.4.20+rocm570-cp39-cp39-manylinux2014_x86_64.whl
Python 3.10
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.20/jaxlib-0.4.20+rocm560-cp310-cp310-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.20/jaxlib-0.4.20+rocm570-cp310-cp310-manylinux2014_x86_64.whl
Python 3.11
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.20/jaxlib-0.4.20+rocm560-cp311-cp311-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.20/jaxlib-0.4.20+rocm570-cp311-cp311-manylinux2014_x86_64.whl
Install Jax
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/archive/refs/tags/jaxlib-v0.4.20.tar.gz
Docker Images
Python 3.9
docker pull rocm/jax:rocm5.6.0-jax0.4.20-py3.9.0
docker pull rocm/jax:rocm5.7.0-jax0.4.20-py3.9.0
Python 3.10
docker pull rocm/jax:rocm5.6.0-jax0.4.20-py3.10.0
docker pull rocm/jax:rocm5.7.0-jax0.4.20-py3.10.0
Python 3.11
docker pull rocm/jax:rocm5.6.0-jax0.4.20-py3.11.0
docker pull rocm/jax:rocm5.7.0-jax0.4.20-py3.11.0
Jax/Jaxlib 0.4.19 release for ROCm
**Jax/Jaxlib 0.4.19 release for ROCm
Install Jaxlib
Python 3.9
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.19/jaxlib-0.4.19+rocm560-cp39-cp39-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.19/jaxlib-0.4.19+rocm570-cp39-cp39-manylinux2014_x86_64.whl
Python 3.10
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.19/jaxlib-0.4.19+rocm560-cp310-cp310-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.19/jaxlib-0.4.19+rocm570-cp310-cp310-manylinux2014_x86_64.whl
Python 3.11
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.19/jaxlib-0.4.19+rocm560-cp311-cp311-manylinux2014_x86_64.whl
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.19/jaxlib-0.4.19+rocm570-cp311-cp311-manylinux2014_x86_64.whl
Install Jax
python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/archive/refs/tags/jaxlib-v0.4.19.tar.gz
Docker Images
Python 3.9
docker pull rocm/jax:rocm5.6.0-jax0.4.19-py3.9.0
docker pull rocm/jax:rocm5.7.0-jax0.4.19-py3.9.0
Python 3.10
docker pull rocm/jax:rocm5.6.0-jax0.4.19-py3.10.0
docker pull rocm/jax:rocm5.7.0-jax0.4.19-py3.10.0
Python 3.11
docker pull rocm/jax:rocm5.6.0-jax0.4.19-py3.11.0
docker pull rocm/jax:rocm5.7.0-jax0.4.19-py3.11.0