jax

Install from the command line


    
  
      
      $ docker pull ghcr.io/nvidia/jax:upstream-pax-2024-11-15

linux/amd64


      
  
      
        $ docker pull ghcr.io/nvidia/jax:upstream-pax-2024-11-15@sha256:d7a1fcc8d660ab07cfb48714dfe4be7fb3c45fa9c9a63cbdb3a57cc5383fd760

linux/arm64


      
  
      
        $ docker pull ghcr.io/nvidia/jax:upstream-pax-2024-11-15@sha256:fc7eb1151c391d0d3c925abab96d4036f95be268f0bc85ed7b2d924765b72ec4

Recent tagged image versions

upstream-pax-2024-11-15 upstream-pax

Published about 20 hours ago · Digest
sha256:7b034319eb1a389ae4182e6db2f62193781971ae13a528daf29d4c2b1356df97

1 Version downloads
jax-2024-11-15 jax

Published about 20 hours ago · Digest
sha256:8bcffb8a767cba64bc104f99294b0534103079dbf2d48c5298134973538b9aad

3 Version downloads
equinox-2024-11-15 equinox

Published about 20 hours ago · Digest
sha256:bd68722304f40593225d285f61e7cf4c5b05db683217d65aa0a3e8ee4bc09636

1 Version downloads
base-2024-11-15 base

Published about 20 hours ago · Digest
sha256:92b746b4e05ec90434e1664869dd07c0cc98890207789a59d52588ee555fe7d2

2 Version downloads
maxtext-2024-11-15 maxtext

Published about 20 hours ago · Digest
sha256:2e31139079a2245b86a3d6ca9ffbb0a47222f95687a498c3d5d27bc59e616de5

1 Version downloads