diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index a99000c2e0..285a06ce31 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -98,7 +98,7 @@ jobs: name: 'JAX' runs-on: ubuntu-latest container: - image: ghcr.io/nvidia/jax:jax + image: ghcr.io/nvidia/jax:jax-2026-07-21 options: --user root steps: - name: 'Dependencies' @@ -149,7 +149,7 @@ jobs: - name: Start named container run: | - docker run -v $(pwd):$(pwd) -w $(pwd) --name builder -d ghcr.io/nvidia/jax:jax sleep infinity + docker run -v $(pwd):$(pwd) -w $(pwd) --name builder -d ghcr.io/nvidia/jax:jax-2026-07-21 sleep infinity - name: 'Dependencies' run: |