Getting Started

Build Pipeline Status

Live build and test status for the published JAX containers
View as Markdown

Workflow status

The table below lists the published containers and their Dockerfiles. Each container is built nightly for amd64 and arm64. For the full, always-current matrix of per-test status badges (single-GPU, multi-GPU, distributed, etc.), see the build-pipeline table in the repository README and the CI workflow runs.

ComponentContainerDockerfile
base = {CUDA, cuDNN, NCCL, OFED, EFA}ghcr.io/nvidia/jax:baseDockerfile.base
core = {base, JAX, Flax, TE}ghcr.io/nvidia/jax:jaxDockerfile.jax
Equinox = {core, Equinox}ghcr.io/nvidia/jax:equinoxDockerfile.equinox
MaxText = {core, MaxText}ghcr.io/nvidia/jax:maxtextDockerfile.maxtext
AXLearn = {core, AXLearn}ghcr.io/nvidia/jax:axlearnDockerfile.axlearn
AlphaFold = {core, AlphaFold}ghcr.io/nvidia/jax:alphafoldDockerfile.alphafold

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.