From f212634454f63021b05af3669e6700609304a213 Mon Sep 17 00:00:00 2001
From: Frederic Bastien <fbastien@nvidia.com>
Date: Thu, 10 Aug 2023 08:29:58 -0700
Subject: [PATCH] Add a link to JAX Toolbox containers

---
 README.md | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/README.md b/README.md
index caf2f5e5a..a35aeb98c 100644
--- a/README.md
+++ b/README.md
@@ -391,6 +391,8 @@ installed as the `jaxlib` package. Use the following instructions to install a
 binary package with `pip` or `conda`, or to [build JAX from
 source](https://jax.readthedocs.io/en/latest/developer.html#building-from-source).
 
+There is also [containers](#containers).
+
 We support installing or building `jaxlib` on Linux (Ubuntu 20.04 or later) and
 macOS (10.12 or later) platforms. There is also *experimental* native Windows
 support.
@@ -583,6 +585,12 @@ See the `conda-forge`
 [jax](https://github.com/conda-forge/jax-feedstock#installing-jax) repositories
 for more details.
 
+### Containers
+
+[JAX Toolbox](https://github.com/NVIDIA/JAX-Toolbox) has bleeding edge
+GPU containers: nightly containers and models/frameworks
+specific containers.
+
 ### Building JAX from source
 See [Building JAX from
 source](https://jax.readthedocs.io/en/latest/developer.html#building-from-source).