Skip to content

Commit 1323056

Browse files
authoredOct 29, 2024··
Update README.md (#1128)
- Added configs tested in internal CI - Added videos section updated with GTC 2024
1 parent 0e4e245 commit 1323056

File tree

1 file changed

+11
-4
lines changed

1 file changed

+11
-4
lines changed
 

‎README.md

+11-4
Original file line numberDiff line numberDiff line change
@@ -333,9 +333,9 @@ We support and test the following JAX frameworks and model architectures. More d
333333
</tbody>
334334
</table>
335335

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.
339339

340340
## Environment Variables
341341

@@ -404,7 +404,14 @@ Docker has traditionally used Docker Schema V2.2 for multi-arch manifest lists b
404404
* [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)
405405

406406
## 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)
408408
* [Slurm and OpenMPI zero config integration](https://jax.readthedocs.io/en/latest/_autosummary/jax.distributed.initialize.html)
409409
* [Adding custom GPU ops](https://jax.readthedocs.io/en/latest/Custom_Operation_for_GPUs.html)
410410
* [Triaging regressions](docs/triage-tool.md)
411+
412+
## Videos
413+
* [Equinox for JAX: The Foundation of an Ecosystem for Science and Machine Learning](https://www.nvidia.com/en-us/on-demand/session/gtc24-s62668/)
414+
* [Scaling Grok with JAX and H100](https://www.nvidia.com/en-us/on-demand/session/gtc24-s63257/)
415+
* [JAX Supercharged on GPUs: High Performance LLMs with JAX and OpenXLA](https://www.nvidia.com/en-us/on-demand/session/gtc24-s62246/)
416+
* [What's New in JAX | GTC Spring 2024](https://www.nvidia.com/en-us/on-demand/session/gtc24-s62659/)
417+
* [What's New in JAX | GTC Spring 2023](https://www.nvidia.com/en-us/on-demand/session/gtcspring23-s51956/)

0 commit comments

Comments
 (0)
Please sign in to comment.