You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardexpand all lines: README.md
+11-4
Original file line number
Diff line number
Diff line change
@@ -333,9 +333,9 @@ We support and test the following JAX frameworks and model architectures. More d
333
333
</tbody>
334
334
</table>
335
335
336
-
In all of the above cases, `ghcr.io/nvidia/jax:XXX` points to the most recent
337
-
nightly build of the container for `XXX`. These containers are also tagged as
338
-
`ghcr.io/nvidia/jax:XXX-YYYY-MM-DD`, if a stable reference is required.
336
+
In all cases, `ghcr.io/nvidia/jax:XXX` points to latest nightly build of the container for `XXX`. For a stable reference, use `ghcr.io/nvidia/jax:XXX-YYYY-MM-DD`.
337
+
338
+
In addition to the public CI, we also run internal CI tests on H100 SXM 80GB and A100 SXM 80GB.
339
339
340
340
## Environment Variables
341
341
@@ -404,7 +404,14 @@ Docker has traditionally used Docker Schema V2.2 for multi-arch manifest lists b
404
404
*[Running a deep learning workload with JAX on multinode multi-GPU clusters on OCI](https://blogs.oracle.com/cloud-infrastructure/post/running-multinode-jax-clusters-on-oci-gpu-cloud)
405
405
406
406
## Resources
407
-
*[What's New in JAX | GTC Spring 2023](https://www.nvidia.com/en-us/on-demand/session/gtcspring23-s51956/)
407
+
*[JAX | NVIDIA NGC Container](https://catalog.ngc.nvidia.com/orgs/nvidia/containers/jax)
408
408
*[Slurm and OpenMPI zero config integration](https://jax.readthedocs.io/en/latest/_autosummary/jax.distributed.initialize.html)
0 commit comments