rocm/jax-training

Verified Publisher

By AMD

•Updated 6 days ago

jax training docker images

Image

10K+

Overview Tags

Tags for rocm/jax-training

Sort by

TAG

base-v26.7

Last pushed 6 days by rocm

docker pull rocm/jax-training:base-v26.7

Digest	OS/ARCH	Compressed size
55d596de328d	linux/amd64	7.5 GB

TAG

latest

Last pushed 6 days by rocm

docker pull rocm/jax-training:latest

Digest	OS/ARCH	Compressed size
b5117e775594	linux/amd64	11.46 GB

TAG

maxtext-v26.7-jax0.11.0-te2.17.0

Last pushed 6 days by rocm

docker pull rocm/jax-training:maxtext-v26.7-jax0.11.0-te2.17.0

Digest	OS/ARCH	Compressed size
b5117e775594	linux/amd64	11.46 GB

TAG

maxtext-v26.7

Last pushed 6 days by rocm

docker pull rocm/jax-training:maxtext-v26.7

Digest	OS/ARCH	Compressed size
b5117e775594	linux/amd64	11.46 GB

TAG

base-v26.6

Last pushed 17 days by rocm

docker pull rocm/jax-training:base-v26.6

Digest	OS/ARCH	Compressed size
0995b15b02ec	linux/amd64	7.35 GB

TAG

maxtext-v26.6-jax0.11.0-te2.17.0

Last pushed 17 days by rocm

docker pull rocm/jax-training:maxtext-v26.6-jax0.11.0-te2.17.0

Digest	OS/ARCH	Compressed size
a71d8dbb045e	linux/amd64	11.19 GB

TAG

maxtext-v26.6

Last pushed 17 days by rocm

docker pull rocm/jax-training:maxtext-v26.6

Digest	OS/ARCH	Compressed size
a71d8dbb045e	linux/amd64	11.19 GB

TAG

maxtext-v26.5-jax0.10.0-te2.15.0

Last pushed about 2 months by rocm

docker pull rocm/jax-training:maxtext-v26.5-jax0.10.0-te2.15.0

Digest	OS/ARCH	Compressed size
21abbf2a1425	linux/amd64	12.09 GB

TAG

maxtext-v26.5

Last pushed about 2 months by rocm

docker pull rocm/jax-training:maxtext-v26.5

Digest	OS/ARCH	Compressed size
21abbf2a1425	linux/amd64	12.09 GB

TAG

maxtext-v26.4

Last pushed about 2 months by rocm

docker pull rocm/jax-training:maxtext-v26.4

Digest	OS/ARCH	Compressed size
9b263b38c159	linux/amd64	15.4 GB

TAG

maxtext-v26.3.2

Last pushed 2 months by rocmdev

docker pull rocm/jax-training:maxtext-v26.3.2

Digest	OS/ARCH	Compressed size
1526f6cbf4fd	linux/amd64	17.88 GB

TAG

maxtext-v25.11-ainic-rccl-rebuild

Last pushed 3 months by rocmdev

docker pull rocm/jax-training:maxtext-v25.11-ainic-rccl-rebuild

Digest	OS/ARCH	Compressed size
8a4e92b36a7e	linux/amd64	25.16 GB

TAG

maxtext-v26.4-jax0.9.1-te2.12.0

Last pushed 3 months by rocmdev

docker pull rocm/jax-training:maxtext-v26.4-jax0.9.1-te2.12.0

Digest	OS/ARCH	Compressed size
9b263b38c159	linux/amd64	15.4 GB

TAG

maxtext-v26.3.1

Last pushed 4 months by rocmdev

docker pull rocm/jax-training:maxtext-v26.3.1

Digest	OS/ARCH	Compressed size
75782ecbc38d	linux/amd64	16.66 GB

TAG

maxtext-v26.3-jax09-EarlyAccess

Last pushed 4 months by rocmdev

docker pull rocm/jax-training:maxtext-v26.3-jax09-EarlyAccess

Digest	OS/ARCH	Compressed size
5d7c0c619bbb	linux/amd64	16.33 GB

TAG

maxtext-v26.3

Last pushed 4 months by rocmdev

docker pull rocm/jax-training:maxtext-v26.3

Digest	OS/ARCH	Compressed size
548381e496fa	linux/amd64	16.66 GB

TAG

maxtext-v26.2

Last pushed 7 months by rocmdev

docker pull rocm/jax-training:maxtext-v26.2

Digest	OS/ARCH	Compressed size
a89643388487	linux/amd64	22.12 GB

TAG

maxtext-v26.1

Last pushed 7 months by rocmdev

docker pull rocm/jax-training:maxtext-v26.1

Digest	OS/ARCH	Compressed size
901083bde353	linux/amd64	21.68 GB

TAG

maxtext-v25.11-ainic

Last pushed 8 months by rocmdev

docker pull rocm/jax-training:maxtext-v25.11-ainic

Digest	OS/ARCH	Compressed size
82b7d6193462	linux/amd64	24.36 GB

TAG

maxtext-v25.9.1

Last pushed 9 months by rocmdev

docker pull rocm/jax-training:maxtext-v25.9.1

Digest	OS/ARCH	Compressed size
60946cfbd470	linux/amd64	20.08 GB

TAG

maxtext-v25.11-jax08-EarlyAccess

Last pushed 9 months by rocmdev

docker pull rocm/jax-training:maxtext-v25.11-jax08-EarlyAccess

Digest	OS/ARCH	Compressed size
0ee8405a7440	linux/amd64	17.95 GB

TAG

maxtext-v25.11

Last pushed 9 months by rocmdev

docker pull rocm/jax-training:maxtext-v25.11

Digest	OS/ARCH	Compressed size
18e4d8f0b8ce	linux/amd64	23.86 GB

TAG

maxtext-v25.9

Last pushed 11 months by rocmdev

docker pull rocm/jax-training:maxtext-v25.9

Digest	OS/ARCH	Compressed size
4bb16ab58279	linux/amd64	16.84 GB

TAG

maxtext-v25.7-jax060

Last pushed about 1 year by rocmdev

docker pull rocm/jax-training:maxtext-v25.7-jax060

Digest	OS/ARCH	Compressed size
7352212ae033	linux/amd64	15.58 GB

TAG

maxtext-v25.7

Last pushed about 1 year by rocmdev

docker pull rocm/jax-training:maxtext-v25.7

Digest	OS/ARCH	Compressed size
45f4c727d401	linux/amd64	10.49 GB