Skip to content

rocm-jaxlib-v0.4.29
Compare
Choose a tag to compare
@zahiqbal zahiqbal released this 15 Jun 23:54
· 3925 commits to main since this release
rocm-jaxlib-v0.4.29
0ab9785

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