jax

Install from the command line


    
  
      
      $ docker pull ghcr.io/nvidia/jax:upstream-t5x-2024-09-19

linux/amd64


      
  
      
        $ docker pull ghcr.io/nvidia/jax:upstream-t5x-2024-09-19@sha256:8ae3367345921f94cb3e62691172a50b5dbd2db6bef8e723dad4629d0e7bb305

linux/arm64


      
  
      
        $ docker pull ghcr.io/nvidia/jax:upstream-t5x-2024-09-19@sha256:fccb9354537d0b8d8c95902607224203957a0ae20d0016b2dc84687ef81d4db9

Recent tagged image versions

upstream-t5x-2024-09-19 upstream-t5x

Published about 17 hours ago · Digest
sha256:26922b8c7492faf9e1c75da1f34458f63d12b6fd11f1ffca9c85d28181f8c396

0 Version downloads
upstream-pax-2024-09-19 upstream-pax

Published about 17 hours ago · Digest
sha256:6fa29330062f1aa23417026903d0c9b492c037cbf8f331bc2f50a61b8cf3f2a3

0 Version downloads
pax-2024-09-19 pax

Published about 17 hours ago · Digest
sha256:6e5b0bf5a6e7c3d4ca29ece3245575c15a43f0198e762e6ce8b5e38dc7cfe787

6 Version downloads
maxtext-2024-09-19 maxtext

Published about 17 hours ago · Digest
sha256:bff64314d300740e38c98de4214f1b0f26b515fa4a687da697e43333c35588f7

15 Version downloads
base-2024-09-19 base

Published about 17 hours ago · Digest
sha256:d325f2d2dc0e3213916dc91fe23566de404a0ee33c71536c3c5759fc87e8b593

3 Version downloads