jax

Install from the command line


    
  
      
      $ docker pull ghcr.io/nvidia/jax:upstream-t5x-2024-08-22

linux/amd64


      
  
      
        $ docker pull ghcr.io/nvidia/jax:upstream-t5x-2024-08-22@sha256:d74d51a05edc36a6d6193bfaa1cf88b3bb26d010159c59d410b5cd2eac4905fb

linux/arm64


      
  
      
        $ docker pull ghcr.io/nvidia/jax:upstream-t5x-2024-08-22@sha256:b50a31c6b365466005b2d9bbb4b096b0a5e0b5368ea599cb4e034d9c9a5789dd

Recent tagged image versions

upstream-t5x-2024-08-22 upstream-t5x

Published about 6 hours ago · Digest
sha256:92659e6ce2f1700fce00a912dcd25fa2f8953a63b5f320237e44be1ecc84dc62

1 Version downloads
upstream-pax-2024-08-22 upstream-pax

Published about 6 hours ago · Digest
sha256:e45d945b2803deeeb2cc176e9b4da20bbf9ca04cf8e4f39fb4dbdd6e77b57523

1 Version downloads
triton-2024-08-22 triton

Published about 6 hours ago · Digest
sha256:00447bb89de317c5587e24c0dda725cf93f3ec433410c0357d42cb09156be0e0

1 Version downloads
maxtext-2024-08-22 maxtext

Published about 6 hours ago · Digest
sha256:f7e1be0d4be0a087cd931acf170066964d1f1d4ca0bf5575662a2b17630cb03a

6 Version downloads
t5x-2024-08-22 t5x

Published about 6 hours ago · Digest
sha256:761387937df539e5d56d7a51af8d208e0c8a9bb1f1644e5a0e9655ffb096316c

7 Version downloads