jax

Install from the command line


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

linux/amd64


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

linux/arm64


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

Recent tagged image versions

upstream-t5x-2024-09-23 upstream-t5x

Published about 2 hours ago · Digest
sha256:6fabd6bbd63e637aedbdbdf68964816ed692b954b685df6ee1a2267dfcb5bdc3

0 Version downloads
t5x-2024-09-23 t5x

Published about 2 hours ago · Digest
sha256:47494ab5caf01729280823f8b9ab9ba6acdb766ca05efe5724340c2f8efe7c99

0 Version downloads
pax-2024-09-23 pax

Published about 2 hours ago · Digest
sha256:911732d442df88cd705c4939c1a372329fc35682ce6707ec20847bddbb20eb29

0 Version downloads
maxtext-2024-09-23 maxtext

Published about 2 hours ago · Digest
sha256:61dba5f53ca57a3284c935820ff5db2c990f8446f3643664f9bdc055d3f4521f

2 Version downloads
levanter-2024-09-23 levanter

Published about 2 hours ago · Digest
sha256:9c551abe5a1d6721daad090aa88b465e208bd36cd978119ab20144bf513fd1d4

1 Version downloads