jax base-scale-training-2026-02-21 Public Latest

Install from the command line


      
  
      
        $ docker pull ghcr.io/nvidia/jax:base-scale-training-2026-02-21

linux/amd64


      
  
      
        $ docker pull ghcr.io/nvidia/jax:base-scale-training-2026-02-21@sha256:9c65656838398c02d0d8370d1db5c717893981cafed439432b7105504b9e3e2f

linux/arm64


      
  
      
        $ docker pull ghcr.io/nvidia/jax:base-scale-training-2026-02-21@sha256:0e695ef7a7e0e14bad84b0a15c12a6c15e2d25aa3c293f59c254119b28d3e69b

Recent tagged image versions

base-scale-training-2026-02-21 base-scale-training

Published 3 days ago · Digest
sha256:a0ff62bb2701499fbc44cba34656eb9c3d813000db99da0de4ae69bccd624494

0 Version downloads
equinox-scale-training-2026-02-21 equinox-scale-training

Published 3 days ago · Digest
sha256:5b0bc36b1539cbf9270d8641e9e5d48a2c4dae46811fee4a158c8ad957989f73

0 Version downloads
jax-scale-training-2026-02-21 jax-scale-training

Published 3 days ago · Digest
sha256:294c2c1f834d0872e5b792c2676db29974dc5bb600faee77b1d39e410e946e29

1 Version downloads
maxtext-scale-training-2026-02-21 maxtext-scale-training

Published 3 days ago · Digest
sha256:34461376d3c4f2ad2f4d807a6c1c9b7c952015d4aa89200d74eef65248e1ca4c

0 Version downloads
maxtext-2026-02-20 maxtext

Published 4 days ago · Digest
sha256:ac9deed07cc715bbe22ee2fafd19a6935772a389fea134b7102c4ed43df0e1b6

8 Version downloads