Help

System theme

Docker Suite

rocm/jax

Verified Publisher

By AMD

•Updated 5 days ago

The release of JAX for ROCm backend.

Image

5

10K+

Tags for rocm/jax

Sort by

TAG

rocm10.1.0_ubuntu26.04_py3.14_jax_release_0.11.1

Last pushed 5 days by rocm

docker pull rocm/jax:rocm10.1.0_ubuntu26.04_py3.14_jax_release_0.11.1

Digest	OS/ARCH	Compressed size
54ea2526d66a	linux/amd64	7.72 GB

TAG

rocm10.1.0_ubuntu26.04_py3.14_jax_release_0.11.0

Last pushed 5 days by rocm

docker pull rocm/jax:rocm10.1.0_ubuntu26.04_py3.14_jax_release_0.11.0

Digest	OS/ARCH	Compressed size
6e39918a1fa4	linux/amd64	7.71 GB

TAG

rocm10.1.0_ubuntu26.04_py3.14_jax_release_0.10.2

Last pushed 5 days by rocm

docker pull rocm/jax:rocm10.1.0_ubuntu26.04_py3.14_jax_release_0.10.2

Digest	OS/ARCH	Compressed size
4b6735e603b7	linux/amd64	7.71 GB

TAG

rocm10.1.0_ubuntu24.04_py3.12_jax_release_0.11.1

Last pushed 5 days by rocm

docker pull rocm/jax:rocm10.1.0_ubuntu24.04_py3.12_jax_release_0.11.1

Digest	OS/ARCH	Compressed size
9bd8fa61eb4d	linux/amd64	7.69 GB

TAG

Last pushed 5 days by rocm

docker pull rocm/jax:latest

Digest	OS/ARCH	Compressed size
9bd8fa61eb4d	linux/amd64	7.69 GB

TAG

rocm10.1.0_ubuntu24.04_py3.12_jax_release_0.11.0

Last pushed 5 days by rocm

docker pull rocm/jax:rocm10.1.0_ubuntu24.04_py3.12_jax_release_0.11.0

Digest	OS/ARCH	Compressed size
8dcec0bd9871	linux/amd64	7.68 GB

TAG

rocm10.1.0_ubuntu24.04_py3.12_jax_release_0.10.2

Last pushed 5 days by rocm

docker pull rocm/jax:rocm10.1.0_ubuntu24.04_py3.12_jax_release_0.10.2

Digest	OS/ARCH	Compressed size
d1690b214154	linux/amd64	7.68 GB

TAG

rocm7.14.1-jax0.10.0-py3.14

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm7.14.1-jax0.10.0-py3.14

Digest	OS/ARCH	Compressed size
78b10efd556f	linux/amd64	7.54 GB

TAG

rocm7.14.1-jax0.10.0-py3.13

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm7.14.1-jax0.10.0-py3.13

Digest	OS/ARCH	Compressed size
fa327c84043e	linux/amd64	7.53 GB

TAG

rocm7.14.1-jax0.10.0-py3.12

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm7.14.1-jax0.10.0-py3.12

Digest	OS/ARCH	Compressed size
37ac40709823	linux/amd64	7.51 GB

TAG

rocm7.14.1-jax0.10.0-py3.11

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm7.14.1-jax0.10.0-py3.11

Digest	OS/ARCH	Compressed size
fe6400020151	linux/amd64	7.54 GB

TAG

rocm7.14.1-jax0.9.1-py3.14

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm7.14.1-jax0.9.1-py3.14

Digest	OS/ARCH	Compressed size
1a9db4ef73af	linux/amd64	7.54 GB

TAG

rocm7.14.1-jax0.9.1-py3.13

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm7.14.1-jax0.9.1-py3.13

Digest	OS/ARCH	Compressed size
2090bb1416d7	linux/amd64	7.54 GB

TAG

rocm7.14.1-jax0.9.1-py3.12

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm7.14.1-jax0.9.1-py3.12

Digest	OS/ARCH	Compressed size
037dfb094838	linux/amd64	7.51 GB

TAG

rocm7.14.1-jax0.9.1-py3.11

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm7.14.1-jax0.9.1-py3.11

Digest	OS/ARCH	Compressed size
6396282dbb57	linux/amd64	7.54 GB

TAG

rocm10.0-jax0.11.0-py3.14

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm10.0-jax0.11.0-py3.14

Digest	OS/ARCH	Compressed size
0535bd55e09c	linux/amd64	7.78 GB

TAG

rocm10.0-jax0.11.0-py3.13

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm10.0-jax0.11.0-py3.13

Digest	OS/ARCH	Compressed size
0369efd4d1da	linux/amd64	7.77 GB

TAG

rocm10.0-jax0.11.0-py3.12

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm10.0-jax0.11.0-py3.12

Digest	OS/ARCH	Compressed size
305ad3e45da0	linux/amd64	7.75 GB

TAG

rocm10.0-jax0.10.2-py3.14

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm10.0-jax0.10.2-py3.14

Digest	OS/ARCH	Compressed size
72e3ae0548d0	linux/amd64	7.77 GB

TAG

rocm10.0-jax0.10.2-py3.13

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm10.0-jax0.10.2-py3.13

Digest	OS/ARCH	Compressed size
67260e61b5fb	linux/amd64	7.77 GB

TAG

rocm10.0-jax0.10.2-py3.12

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm10.0-jax0.10.2-py3.12

Digest	OS/ARCH	Compressed size
e411002c2324	linux/amd64	7.74 GB

TAG

rocm10.0-jax0.10.2-py3.11

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm10.0-jax0.10.2-py3.11

Digest	OS/ARCH	Compressed size
f4042c52678c	linux/amd64	7.77 GB

TAG

rocm10.0-jax0.10.1-py3.14

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm10.0-jax0.10.1-py3.14

Digest	OS/ARCH	Compressed size
0be939ae01dd	linux/amd64	7.79 GB

TAG

rocm10.0-jax0.10.1-py3.13

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm10.0-jax0.10.1-py3.13

Digest	OS/ARCH	Compressed size
8597e93f6d6b	linux/amd64	7.78 GB

TAG

rocm10.0-jax0.10.1-py3.12

Last pushed about 1 month by rocm

docker pull rocm/jax:rocm10.0-jax0.10.1-py3.12

Digest	OS/ARCH	Compressed size
d0609b6fa4fd	linux/amd64	7.76 GB