Sign inSign up

rocm/jax-training

Verified Publisher

By AMD

•Updated 19 days ago

jax training docker images

Image
0

10K+

Tags for rocm/jax-training

Sort by

TAG

latest

Last pushed 20 days by rocm

docker pull rocm/jax-training:latest
DigestOS/ARCHCompressed size

21abbf2a1425

linux/amd64

12.09 GB

TAG

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

Last pushed 20 days by rocm

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

21abbf2a1425

linux/amd64

12.09 GB

TAG

maxtext-v26.5

Last pushed 20 days by rocm

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

21abbf2a1425

linux/amd64

12.09 GB

TAG

maxtext-v26.4

Last pushed 20 days by rocm

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

9b263b38c159

linux/amd64

15.4 GB

TAG

maxtext-v26.3.2

Last pushed about 1 month by rocmdev

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

1526f6cbf4fd

linux/amd64

17.88 GB

TAG

maxtext-v25.11-ainic-rccl-rebuild

Last pushed about 2 months by rocmdev

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

8a4e92b36a7e

linux/amd64

25.16 GB

TAG

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

Last pushed 2 months by rocmdev

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

9b263b38c159

linux/amd64

15.4 GB

TAG

maxtext-v26.3.1

Last pushed 2 months by rocmdev

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

75782ecbc38d

linux/amd64

16.66 GB

TAG

maxtext-v26.3-jax09-EarlyAccess

Last pushed 3 months by rocmdev

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

5d7c0c619bbb

linux/amd64

16.33 GB

TAG

maxtext-v26.3

Last pushed 3 months by rocmdev

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

548381e496fa

linux/amd64

16.66 GB

TAG

maxtext-v26.2

Last pushed 6 months by rocmdev

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

a89643388487

linux/amd64

22.12 GB

TAG

maxtext-v26.1

Last pushed 6 months by rocmdev

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

901083bde353

linux/amd64

21.68 GB

TAG

maxtext-v25.11-ainic

Last pushed 7 months by rocmdev

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

82b7d6193462

linux/amd64

24.36 GB

TAG

maxtext-v25.9.1

Last pushed 8 months by rocmdev

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

60946cfbd470

linux/amd64

20.08 GB

TAG

maxtext-v25.11-jax08-EarlyAccess

Last pushed 8 months by rocmdev

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

0ee8405a7440

linux/amd64

17.95 GB

TAG

maxtext-v25.11

Last pushed 8 months by rocmdev

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

18e4d8f0b8ce

linux/amd64

23.86 GB

TAG

maxtext-v25.9

Last pushed 10 months by rocmdev

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

4bb16ab58279

linux/amd64

16.84 GB

TAG

maxtext-v25.7-jax060

Last pushed 11 months by rocmdev

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

7352212ae033

linux/amd64

15.58 GB

TAG

maxtext-v25.7

Last pushed 11 months by rocmdev

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

45f4c727d401

linux/amd64

10.49 GB

TAG

maxtext-v25.5

Last pushed over 1 year by rocmdev

docker pull rocm/jax-training:maxtext-v25.5
DigestOS/ARCHCompressed size

4e0516358a22

linux/amd64

9.49 GB

TAG

maxtext-v25.4

Last pushed over 1 year by rocmdev

docker pull rocm/jax-training:maxtext-v25.4
DigestOS/ARCHCompressed size

fb3eb71cd742

linux/amd64

9.91 GB