Set NCCL env profile

aa872a9

Select commit

Loading

Failed to load commit list.

Uh oh!

There was an error while loading. Please reload this page.

jax-toolbox-triage pure-Python CI on: pull_request

mypy

pytest (3.8)

pytest (3.14)

ruff (check)

ruff (format)

nsys-jax non-GPU CI on: pull_request 4

mypy

combine

notebook

ruff

installation (nvidia/cuda:12.6.3-base-ubuntu24.04, cuda-nsight-systems-12-6)

installation (nvidia/cuda:12.8.0-base-ubuntu24.04, cuda-nsight-systems-12-8)

installation (nvidia/cuda:13.1.0-base-ubuntu24.04, cuda-nsight-systems-13-1)

CI on: push 38

metadata

bump-manifest

amd64 / build-base / build-base

amd64 / test-nccl / nccl-test-gke / build-nccl-gke

amd64 / test-nccl / nccl-test-gke / nccl-gke (all_gather_perf_mpi)

amd64 / test-nccl / nccl-test-gke / nccl-gke (all_reduce_perf_mpi)

amd64 / test-nccl / nccl-test-gke / nccl-gke (broadcast_perf_mpi)

amd64 / test-nccl / nccl-test-gke / nccl-gke (reduce_scatter_perf_mpi)

amd64 / test-nccl / build-mpi-operator-compatible-base

amd64 / test-nccl / nccl-test (all_gather_perf_mpi)

amd64 / test-nccl / nccl-test (all_reduce_perf_mpi)

amd64 / test-nccl / nccl-test (broadcast_perf_mpi)

amd64 / test-nccl / nccl-test (reduce_scatter_perf_mpi)

amd64 / build-jax

amd64 / build-equinox

amd64 / build-maxtext

amd64 / build-torchax

amd64 / build-axlearn

amd64 / build-alphafold

amd64 / test-jax (backend-independent, 1)

amd64 / test-jax (single-gpu, 1)

amd64 / test-jax (multi-gpu, 8)

amd64 / test-te (L0_jax_unittest, 8)

amd64 / test-te (L0_jax_distributed_unittest, 8)

amd64 / test-nsys-jax-unit (8)

amd64 / test-nsys-jax-nccl (2, 8)

amd64 / test-cutlass (1, 8)

amd64 / test-maxtext (single-node, 1, 8, 1, 1, 2, 4, NCCL_NET_PLUGIN=none NCCL_NET=Socket)

amd64 / test-maxtext (multi-node, 2, 8, 1, 4, 2, 2, OFI_NCCL_PROTOCOL=SENDRECV)

amd64 / test-maxtext-gke / maxtext-gke (maxtext-dot-product-gke, 2, 8, llama2-7b, 1, 1, 2, 4, dot_product, minimal_with_cont...

amd64 / test-maxtext-gke / maxtext-gke (maxtext-cudnn-flash-te-gke, 2, 8, llama2-7b, 1, 1, 2, 4, cudnn_flash_te, minimal_wit...

amd64 / test-axlearn-eks

amd64 / test-axlearn-fuji-models-eks

amd64 / collect-docker-tags

amd64 / test-alphafold-inference

amd64 / test-nsys-jax-unit-archive

arm64 / build-base / build-base

arm64 / test-nccl

arm64 / build-jax

arm64 / build-equinox

arm64 / build-maxtext

arm64 / build-torchax

arm64 / build-axlearn

arm64 / build-alphafold

arm64 / test-jax

arm64 / test-te

arm64 / test-nsys-jax-unit

arm64 / test-nsys-jax-nccl

arm64 / test-cutlass

arm64 / test-maxtext

arm64 / test-maxtext-gke

arm64 / test-axlearn-eks

arm64 / test-axlearn-fuji-models-eks

arm64 / collect-docker-tags

arm64 / test-alphafold-inference

arm64 / test-nsys-jax-unit-archive

merge-new-manifest

make-publish-configs

publish-containers (jax, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:27015572334-j... / publish

publish-containers (equinox, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:270155723... / publish

publish-containers (maxtext, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:270155723... / publish

publish-containers (axlearn, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:270155723... / publish

publish-containers (base, mock-jax, 800, ghcr.io/nvidia/jax-toolbox-internal:27015572334-base-amd... / publish

publish-containers (jax, mock-jax, 1000, ghcr.io/nvidia/jax-toolbox-internal:27015572334-jax-amd6... / publish

publish-containers (equinox, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:27015572334-equin... / publish

publish-containers (maxtext, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:27015572334-maxte... / publish

publish-containers (axlearn, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:27015572334-axlea... / publish

finalize / workflow-badge

finalize / report

finalize / upload-badge

finalize / publish-badge

Select a check to view from the sidebar