jax

Install from the command line


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

linux/amd64


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

linux/arm64


      
  
      
        $ docker pull ghcr.io/nvidia/jax:upstream-t5x-2024-08-05@sha256:791d2d632c94ea595e43a8030ca88063cb6117918682041f82b304b8fe51b763

Recent tagged image versions

upstream-t5x-2024-08-05 upstream-t5x

Published about 23 hours ago · Digest
sha256:af11fe74f3ccefbb9d08eb41ab98134abebe924349b8dc693a930892c2604116

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

Published about 23 hours ago · Digest
sha256:c7bad5549610d1942a2a7b9240ce0173f645d508fc2f534146c7cc7e1abad963

1 Version downloads
maxtext-2024-08-05 maxtext

Published about 23 hours ago · Digest
sha256:db001a8b7cb0ba0bf9c7d0ca52d1bb9099802bd940600dadff43a5cf6eb5709c

14 Version downloads
equinox-2024-08-05 equinox

Published about 23 hours ago · Digest
sha256:ab52b68401065fb7152278818cbdee03147afb4d5b3e94915b772f99548c5a18

1 Version downloads
pax-2024-08-05 pax

Published about 23 hours ago · Digest
sha256:631a16940712d4ba5891cab6db5de444c727cda92afc9f744b1505f0aeb267ce

11 Version downloads