Add JetStream to MaxText container #3352

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
metadata	1s
bump-manifest	22s
amd64 / test-distribution (extra-only-distribution.sh)	5s
arm64 / test-distribution (extra-only-distribution.sh)	6s
amd64 / test-distribution (mirror-only-distribution.sh)	6s
amd64 / test-distribution (upstream-only-distribution.sh)	4s
arm64 / test-distribution (mirror-only-distribution.sh)	6s
amd64 / test-distribution (local-patch-distribution.sh)	5s
arm64 / test-distribution (upstream-only-distribution.sh)	9s
arm64 / test-distribution (local-patch-distribution.sh)	5s
amd64 / build-base / build-base	3m 42s
arm64 / build-base / build-base	4m 23s
amd64 / build-jax / build-jax	9m 6s
arm64 / build-jax / build-jax	15m 14s
amd64 / build-maxtext / build-maxtext	4m 31s
amd64 / build-levanter / build-levanter	4m 14s
amd64 / build-equinox / build-equinox	3m 13s
amd64 / build-triton / build-triton	11m 37s
amd64 / build-upstream-pax / build-upstream-pax	6m 8s
amd64 / build-gemma / build-gemma	5m 20s
amd64 / build-upstream-t5x / build-upstream-t5x	6m 35s
amd64 / test-jax / jax-V100-unit-test	20m 50s
amd64 / test-jax / jax-A100-unit-test	11m 54s
amd64 / test-nsys-jax / nsys-jax-V100-unit-test	6m 29s
amd64 / test-nsys-jax / nsys-jax-A100-unit-test	9m 0s
amd64 / test-pallas / pallas-V100-unit-test	2m 26s
amd64 / test-pallas / pallas-A100-unit-test	4m 42s
amd64 / test-jax / runner / launch-slurm-runner	5h 18m 52s
amd64 / test-nsys-jax / runner / launch-slurm-runner	5h 20m 27s
amd64 / test-pallas / runner / launch-slurm-runner	5h 24m 17s
amd64 / test-maxtext	0s
amd64 / test-jetstream	0s
amd64 / test-levanter / levanter-V100-unit-test	2m 50s
amd64 / test-levanter / levanter-A100-unit-test	2m 56s
amd64 / test-levanter / runner / launch-slurm-runner	5h 17m 26s
arm64 / build-triton	0s
arm64 / build-gemma	0s
arm64 / test-jax	0s
arm64 / test-nsys-jax	0s
arm64 / test-pallas	0s
arm64 / build-equinox / build-equinox	4m 17s
arm64 / build-levanter / build-levanter	5m 55s
arm64 / build-maxtext / build-maxtext	8m 57s
arm64 / build-upstream-pax / build-upstream-pax	26m 18s
arm64 / build-upstream-t5x / build-upstream-t5x	26m 50s
arm64 / test-triton	0s
arm64 / test-gemma	0s
amd64 / test-gemma / gemma-V100-unit-test	3m 55s
amd64 / test-gemma / gemma-A100-unit-test	5m 26s
amd64 / test-gemma / runner / launch-slurm-runner	5h 22m 33s
amd64 / build-rosetta-pax / build-rosetta	6m 27s
amd64 / test-te / te-V100-unit-test	57m 15s
amd64 / test-te / te-A100-unit-test	19m 38s
amd64 / test-upstream-pax / single-process-multi-device (1, 8, 1, 1)	5h 21m 27s
amd64 / test-upstream-pax / single-process-multi-device (1, 1, 2, 4)	5h 29m 38s
amd64 / test-upstream-pax / single-process-evaluation (1, 8, 1, 1)	5h 51m 10s
amd64 / test-upstream-pax / pax-multi-node (1DP1FSDP1TP1PP, 1, 1, 1, 1, 4)	5h 54m 9s
amd64 / test-upstream-pax / pax-multi-node (8DP1FSDP1TP1PP, 1, 8, 1, 1, 4)	5h 27m 35s
amd64 / test-upstream-pax / pax-multi-node (1DP8FSDP1TP1PP, 1, 1, 8, 1, 4)	5h 24m 36s
amd64 / test-upstream-pax / pax-multi-node (2DP1FSDP1TP4PP, 4, 2, 1, 1, 4)	5h 46m 30s
amd64 / test-upstream-pax / pax-multi-node (4DP1FSDP2TP1PP, 1, 4, 1, 2, 4)	5h 57m 19s
amd64 / test-upstream-pax / pax-multi-node (16DP1FSDP1TP1PP, 1, 16, 1, 1, 4)	5h 41m 45s
amd64 / test-upstream-pax / pax-multi-node (2DP1FSDP2TP4PP, 4, 2, 1, 2, 4)	5h 37m 42s
amd64 / test-upstream-pax / pax-multi-node (LLaMA_eval, 1, 1, 8, 1, 4, true, --model-type LLaMA70BProxy --evaluate)	5h 54m 17s
amd64 / test-te / runner / launch-slurm-runner	6h 0m 14s
amd64 / test-te-multigpu / te-multi-gpu (2) / te-2GPU	3h 45m 50s
amd64 / test-te-multigpu / te-multi-gpu (4) / te-4GPU	3h 42m 25s
amd64 / test-te-multigpu / te-multi-gpu (8) / te-8GPU	3h 39m 21s
amd64 / test-upstream-t5x / t5x-multi-node (8G2N, 8, 2)	6h 0m 14s
amd64 / test-upstream-t5x / t5x-multi-gpu (1P8G, 8)	6h 0m 12s
amd64 / test-upstream-t5x / t5x-multi-node (8G2N_fmha, 8, 2, --enable-fmha 1)	6h 0m 15s
amd64 / test-upstream-t5x / t5x-multi-gpu (1P2G_fmha, 2, --enable-fmha 1)	6h 2m 30s
amd64 / build-rosetta-t5x / build-rosetta	7m 56s
amd64 / test-triton / triton-V100-unit-test	4m 4s
amd64 / test-triton / triton-A100-unit-test	15m 28s
amd64 / test-triton / runner / launch-slurm-runner	6h 0m 11s
arm64 / test-levanter	0s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1DP1FSDP1TP1PP_TE, 1, 1, 1, 1, 4)	6h 0m 14s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (8DP1FSDP1TP1PP_TE, 1, 8, 1, 1, 4)	6h 0m 19s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (1DP8FSDP1TP1PP_TE, 1, 1, 8, 1, 4)	6h 0m 17s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (4DP1FSDP2TP1PP_TE, 1, 4, 1, 2, 4)	6h 0m 4s
amd64 / test-rosetta-pax / rosetta-pax-single-node-dropout-te (1, 8, 1, 1)	6h 0m 16s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (16DP1FSDP1TP1PP_TE, 1, 16, 1, 1, 4)	6h 0m 15s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (5B_fused_attn_1, 1, 1, 8, 1, 2, --model-type 5B)	6h 0m 17s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (5B_fused_attn_0, 1, 1, 8, 1, 2, --model-type 5B --disable-fused-attn)	6h 0m 16s
amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (LLaMA_eval_TE, 1, 1, 8, 1, 4, true, --model-type LLaMA70BProxy --evalu...	6h 0m 4s
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 8, 1, 1)	6h 0m 13s
amd64 / test-rosetta-pax / single-process-multi-device-te (1, 1, 2, 4)	6h 0m 14s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (1, 8, 1, 1)	6h 0m 16s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (1, 4, 1, 2)	6h 0m 16s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (4, 2, 1, 1)	6h 0m 19s
amd64 / test-rosetta-pax / rosetta-pax-multi-node (4, 2, 1, 2)	6h 0m 14s
amd64 / test-rosetta-pax / single-process-evaluation-te (1, 8, 1, 1)	6h 0m 5s
amd64 / collect-docker-tags	1s
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N8G-te-1, 8, 2, --gin.train/utils.DatasetConfig.pack=False --gin.train_eva...	6h 0m 16s
amd64 / test-rosetta-t5x / single-process-multi-device (1P1G_te-0, 1, --enable-te 0)	6h 0m 15s
amd64 / test-rosetta-t5x / single-process-multi-device (1P8G_te-1, 8, --gin.train/utils.DatasetConfig.pack=False --gin.train...	6h 0m 16s
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N2G_te-0, 2, 2, --enable-te 0)	6h 0m 4s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 1)	6h 0m 13s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 2)	6h 0m 9s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 1)	6h 0m 14s
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 2)	6h 0m 12s
amd64 / test-rosetta-t5x / vit-single-process-multi-device (8)	6h 0m 2s
arm64 / test-maxtext	0s
arm64 / test-jetstream	0s
arm64 / test-upstream-pax	0s
arm64 / test-te	0s
arm64 / test-te-multigpu	0s
arm64 / build-rosetta-pax / build-rosetta	11m 40s
arm64 / test-upstream-t5x	0s
arm64 / build-rosetta-t5x / build-rosetta	14m 8s
arm64 / test-rosetta-pax	0s
arm64 / collect-docker-tags	1s
arm64 / test-rosetta-t5x	0s
amd64 / test-te-multigpu / sitrep	4s
amd64 / test-upstream-pax / test-upstream-pax-summary	1s
amd64 / test-upstream-pax / test-upstream-pax-metrics	26s
amd64 / test-upstream-pax / test-upstream-pax-sitrep / sitrep	22s
amd64 / test-upstream-pax / test-upstream-pax-outcome	1s
amd64 / test-upstream-t5x / test-upstream-t5x-summary	1s
amd64 / test-upstream-t5x / test-upstream-t5x-metrics	0s
amd64 / test-upstream-t5x / test-upstream-t5x-sitrep / sitrep	7s
amd64 / test-upstream-t5x / test-upstream-t5x-outcome	2s
amd64 / test-rosetta-pax / test-pax-rosetta-summary	1s
amd64 / test-rosetta-pax / test-pax-rosetta-metrics	0s
amd64 / test-rosetta-pax / test-pax-rosetta-sitrep / sitrep	7s
amd64 / test-rosetta-pax / test-pax-rosetta-outcome	1s
amd64 / test-rosetta-t5x / test-t5x-rosetta-summary	1s
amd64 / test-rosetta-t5x / test-t5x-rosetta-metrics	0s
amd64 / test-rosetta-t5x / test-t5x-rosetta-sitrep / sitrep	17s
amd64 / test-rosetta-t5x / test-t5x-rosetta-outcome	1s
make-publish-configs	4s
merge-new-manifest	0s
publish-containers (equinox, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:110428597... / publish	19s
publish-containers (base, mock-jax, 800, ghcr.io/nvidia/jax-toolbox-internal:11042859712-base-amd... / publish	25s
publish-containers (equinox, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:11042859712-equin... / publish	23s
publish-containers (gemma, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:11042859712-gemma-a... / publish	13s
publish-containers (jax, mock-jax, 1000, ghcr.io/nvidia/jax-toolbox-internal:11042859712-jax-amd6... / publish	30s
publish-containers (gemma, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:11042859712... / publish	16s
publish-containers (jax, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:11042859712-j... / publish	15s
publish-containers (maxtext, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:110428597... / publish	12s
publish-containers (levanter, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:11042859712-leva... / publish	15s
publish-containers (levanter, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:11042859... / publish	26s
publish-containers (pax, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:11042859712-p... / publish	19s
publish-containers (pax, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:11042859712-pax-amd64... / publish	19s
publish-containers (t5x, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:11042859712-t5x-amd64... / publish	13s
publish-containers (t5x, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:11042859712-t... / publish	26s
publish-containers (triton, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:1104285971... / publish	21s
publish-containers (triton, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:11042859712-triton... / publish	13s
publish-containers (upstream-pax, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:11042859712-... / publish	33s
publish-containers (upstream-pax, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:1104... / publish	21s
publish-containers (upstream-t5x, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:11042859712-... / publish	16s
publish-containers (upstream-t5x, mock-jax-mealkit, 500, ghcr.io/nvidia/jax-toolbox-internal:1104... / publish	14s
finalize / workflow-badge	6s
finalize / report	14s
finalize / upload-badge	15s
finalize / publish-badge	2s
	12d 4h 29m 32s

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Add JetStream to MaxText container #3352

Usage

Add JetStream to MaxText container #3352

Jobs

Run details

Run time