Skip to content

Jax/Jaxlib 0.4.16 release for ROCm

Choose a tag to compare

View all tags
@rahulbatra85 rahulbatra85 released this 11 Oct 16:12
· 1 commit to rocm-jaxlib-v0.4.16 since this release
jaxlib-v0.4.16
e05a86a

**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
Assets 8
Loading