Skip to content

Jax/Jaxlib 0.4.13 release for ROCm 5.6.0
Compare
Choose a tag to compare
@rahulbatra85 rahulbatra85 released this 02 Aug 15:16
jaxlib-v0.4.13-rocm560
9d98e21

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
Assets 5
Loading