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

Jax/Jaxlib 0.4.18 release for ROCm

**Jax/Jaxlib 0.4.18 release for ROCm

Install Jaxlib
Python 3.9

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.18/jaxlib-0.4.18+rocm560-cp39-cp39-manylinux2014_x86_64.whl

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.18/jaxlib-0.4.18+rocm570-cp39-cp39-manylinux2014_x86_64.whl

Python 3.10

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.18/jaxlib-0.4.18+rocm560-cp310-cp310-manylinux2014_x86_64.whl

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.18/jaxlib-0.4.18+rocm570-cp310-cp310-manylinux2014_x86_64.whl

Python 3.11

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.18/jaxlib-0.4.18+rocm560-cp311-cp311-manylinux2014_x86_64.whl

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.18/jaxlib-0.4.18+rocm570-cp311-cp311-manylinux2014_x86_64.whl

Install Jax

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/archive/refs/tags/jaxlib-v0.4.18.tar.gz

Docker Images
Python 3.9

docker pull rocm/jax:rocm5.6.0-jax0.4.18-py3.9.0

docker pull rocm/jax:rocm5.7.0-jax0.4.18-py3.9.0

Python 3.10

docker pull rocm/jax:rocm5.6.0-jax0.4.18-py3.10.0

docker pull rocm/jax:rocm5.7.0-jax0.4.18-py3.10.0

Python 3.11

docker pull rocm/jax:rocm5.6.0-jax0.4.18-py3.11.0

docker pull rocm/jax:rocm5.7.0-jax0.4.18-py3.11.0

Jax/Jaxlib 0.4.17 release for ROCm

**Jax/Jaxlib 0.4.17 release for ROCm

Install Jaxlib
Python 3.9

https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.17/jaxlib-0.4.17+rocm560-cp390-cp390-manylinux2014_x86_64.whl

https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.17/jaxlib-0.4.17+rocm570-cp390-cp390-manylinux2014_x86_64.whl

Python 3.10

https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.17/jaxlib-0.4.17+rocm560-cp310-cp310-manylinux2014_x86_64.whl

https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.17/jaxlib-0.4.17+rocm570-cp310-cp310-manylinux2014_x86_64.whl

Python 3.11

https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.17/jaxlib-0.4.17+rocm560-cp311-cp311-manylinux2014_x86_64.whl

https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.17/jaxlib-0.4.17+rocm570-cp311-cp311-manylinux2014_x86_64.whl

Install Jax

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/archive/refs/tags/jaxlib-v0.4.17.tar.gz

Docker Images
Python 3.9

docker pull rocm/jax:rocm5.6.0-jax0.4.17-py3.9.0

docker pull rocm/jax:rocm5.7.0-jax0.4.17-py3.9.0

Python 3.10

docker pull rocm/jax:rocm5.6.0-jax0.4.17-py3.10.0

docker pull rocm/jax:rocm5.7.0-jax0.4.17-py3.10.0

Python 3.11

docker pull rocm/jax:rocm5.6.0-jax0.4.17-py3.11.0

docker pull rocm/jax:rocm5.7.0-jax0.4.17-py3.11.0

Jax/Jaxlib 0.4.16 release for ROCm

**Jax/Jaxlib 0.4.16 release for ROCm

Install Jaxlib
Python 3.9

https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.16/jaxlib-0.4.16+rocm550-cp39-cp39-manylinux2014_x86_64.whl

https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.16/jaxlib-0.4.16+rocm560-cp39-cp39-manylinux2014_x86_64.whl

Python 3.10

https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.16/jaxlib-0.4.16+rocm550-cp310-cp310-manylinux2014_x86_64.whl

https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.16/jaxlib-0.4.16+rocm560-cp39-cp39-manylinux2014_x86_64.whl

Python 3.11

https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.16/jaxlib-0.4.16+rocm550-cp311-cp311-manylinux2014_x86_64.whl

https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.16/jaxlib-0.4.16+rocm560-cp311-cp311-manylinux2014_x86_64.whl

Install Jax

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/archive/refs/tags/jaxlib-v0.4.16.tar.gz

Docker Images
Python 3.9

docker pull rocm/jax:rocm5.5.0-jax0.4.16-py3.9.0

docker pull rocm/jax:rocm5.6.0-jax0.4.16-py3.9.0

Python 3.10

docker pull rocm/jax:rocm5.5.0-jax0.4.16-py3.10.0

docker pull rocm/jax:rocm5.6.0-jax0.4.16-py3.10.0

Python 3.11

docker pull rocm/jax:rocm5.5.0-jax0.4.16-py3.11.0

docker pull rocm/jax:rocm5.6.0-jax0.4.16-py3.11.0

Jax/Jaxlib 0.4.14 release for ROCm

**Jax/Jaxlib 0.4.14 release for ROCm

Install Jaxlib
Python 3.9

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.14/jaxlib-0.4.14+rocm550-cp390-cp390-manylinux2014_x86_64.whl

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.14/jaxlib-0.4.14+rocm560-cp390-cp390-manylinux2014_x86_64.whl

Python 3.10

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.14/jaxlib-0.4.14+rocm550-cp310-cp310-manylinux2014_x86_64.whl

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.14/jaxlib-0.4.14+rocm560-cp310-cp310-manylinux2014_x86_64.whl

Python 3.11

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.14/jaxlib-0.4.14+rocm550-cp311-cp311-manylinux2014_x86_64.whl

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.14/jaxlib-0.4.14+rocm560-cp311-cp311-manylinux2014_x86_64.whl

Install Jax

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/archive/refs/tags/jaxlib-v0.4.14.tar.gz

Docker Images
Python 3.9

docker pull rocm/jax:rocm5.5.0-jax0.4.14-py3.9.0

docker pull rocm/jax:rocm5.6.0-jax0.4.14-py3.9.0

Python 3.10

docker pull rocm/jax:rocm5.5.0-jax0.4.14-py3.10.0

docker pull rocm/jax:rocm5.6.0-jax0.4.14-py3.10.0

Python 3.11

docker pull rocm/jax:rocm5.5.0-jax0.4.13-py3.11.0

docker pull rocm/jax:rocm5.6.0-jax0.4.14-py3.11.0

Jax/Jaxlib 0.4.13 release for ROCm 5.6.0

Jax/Jaxlib 0.4.13 release for ROCm 5.6.0

Install Jaxlib
Python 3.8

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.13-rocm560/jaxlib-0.4.13+rocm560-cp38-cp38-manylinux2014_x86_64.whl

Python 3.9

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.13-rocm560/jaxlib-0.4.13+rocm560-cp39-cp39-manylinux2014_x86_64.whl

Python 3.10

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/releases/download/jaxlib-v0.4.13-rocm560/jaxlib-0.4.13+rocm560-cp310-cp310-manylinux2014_x86_64.whl

Install Jax

python3 -m pip install https://github.com/ROCmSoftwarePlatform/jax/archive/refs/tags/jaxlib-v0.4.13-rocm560.tar.gz

Docker Images
Python 3.8

docker pull rocm/jax:rocm5.6.0-jax0.4.13-py3.8.0

Python 3.9

docker pull rocm/jax:rocm5.6.0-jax0.4.13-py3.9.0

Python 3.10

docker pull rocm/jax:rocm5.6.0-jax0.4.13-py3.10.0