> For clean Markdown content of this page, append .md to this URL. For the complete documentation index, see https://docs.nvidia.com/jax-toolbox/llms.txt

# Introduction

JAX Toolbox provides a public CI, Docker images for popular JAX libraries, and
optimized JAX examples to simplify and enhance your JAX development experience on
NVIDIA GPUs. It supports JAX libraries such as
[MaxText](https://github.com/google/maxtext) and
[Pallas](https://jax.readthedocs.io/en/latest/pallas/quickstart.html).

## Where to go next

- [Frameworks & Supported Models](/jax-toolbox/frameworks) — the frameworks and
  model architectures we support and test, and their containers.
- [Build Pipeline Status](/jax-toolbox/build-status) — live build and test
  status for every published container.
- [Environment Variables](/jax-toolbox/environment-variables) — the XLA and
  NCCL flags baked into the images for performance tuning.
- [Profiling](/jax-toolbox/performance-profiling/profiling) — how to profile JAX programs on GPU.
- [Container Versions](/jax-toolbox/container-versions) — base-container history and
  staging releases.

## Containers

In all cases, `ghcr.io/nvidia/jax:XXX` points to the latest nightly build of the
container for `XXX`. For a stable reference, use `ghcr.io/nvidia/jax:XXX-YYYY-MM-DD`.

In addition to the public CI, we also run internal CI nightlies on GB300, B300,
GB200, B200, DGX Spark, RTX PRO 6000 Blackwell, Jetson AGX Thor, H100 SXM 80GB,
and A100 SXM 80GB.

## License

JAX Toolbox is released under the
[Apache 2.0 License](https://github.com/NVIDIA/JAX-Toolbox/blob/main/LICENSE.md).