Skip to content

Merge branch 'main' into aybchan/update-readme-structure #6024

Merge branch 'main' into aybchan/update-readme-structure

Merge branch 'main' into aybchan/update-readme-structure #6024

Triggered via push June 26, 2026 11:34
Status Failure
Total duration 3h 21m 6s
Artifacts 95

ci.yaml

on: push
metadata
4s
metadata
bump-manifest
29s
bump-manifest
amd64  /  ...  /  build-base
1m 59s
amd64 / build-base / build-base
arm64  /  ...  /  build-base
2m 26s
arm64 / build-base / build-base
amd64  /  ...  /  build-mpi-operator-compatible-base
4m 29s
amd64 / test-nccl / build-mpi-operator-compatible-base
amd64  /  ...  /  build-nccl-gke
2m 51s
amd64 / test-nccl / nccl-test-gke / build-nccl-gke
arm64  /  ...  /  build-mpi-operator-compatible-base
arm64 / test-nccl / build-mpi-operator-compatible-base
arm64  /  ...  /  build-nccl-gke
arm64 / test-nccl / nccl-test-gke / build-nccl-gke
Matrix: amd64 / test-cutlass
Matrix: amd64 / test-jax
Matrix: amd64 / test-nsys-jax-nccl
Matrix: amd64 / test-te
Matrix: amd64 / test-nsys-jax-unit
Matrix: amd64 / test-nccl / nccl-test
Matrix: amd64 / test-nccl / nccl-test-gke / nccl-gke
Matrix: arm64 / test-cutlass
Matrix: arm64 / test-jax
Matrix: arm64 / test-nsys-jax-nccl
Matrix: arm64 / test-te
Matrix: arm64 / test-nsys-jax-unit
Matrix: arm64 / test-nccl / nccl-test
Waiting for pending jobs
Matrix: arm64 / test-nccl / nccl-test-gke / nccl-gke
Waiting for pending jobs
Matrix: amd64 / test-maxtext-gke / maxtext-gke
Matrix: amd64 / test-maxtext
amd64  /  test-alphafold-inference
5m 19s
amd64 / test-alphafold-inference
amd64  /  test-axlearn-eks
16m 14s
amd64 / test-axlearn-eks
amd64  /  test-axlearn-fuji-models-eks
2m 43s
amd64 / test-axlearn-fuji-models-eks
Matrix: amd64 / test-nsys-jax-unit-archive
amd64  /  collect-docker-tags
2s
amd64 / collect-docker-tags
Matrix: arm64 / test-maxtext-gke / maxtext-gke
Waiting for pending jobs
Matrix: arm64 / test-maxtext
arm64  /  test-alphafold-inference
arm64 / test-alphafold-inference
arm64  /  test-axlearn-eks
arm64 / test-axlearn-eks
arm64  /  test-axlearn-fuji-models-eks
arm64 / test-axlearn-fuji-models-eks
Matrix: arm64 / test-nsys-jax-unit-archive
arm64  /  collect-docker-tags
5s
arm64 / collect-docker-tags
make-publish-configs
3s
make-publish-configs
merge-new-manifest
0s
merge-new-manifest
Matrix: publish-containers
finalize  /  workflow-badge
7s
finalize / workflow-badge
finalize  /  report
14s
finalize / report
finalize  /  upload-badge
12s
finalize / upload-badge
finalize  /  publish-badge
4s
finalize / publish-badge
Fit to window
Zoom out
Zoom in

Annotations

7 errors and 31 warnings
amd64 / test-nsys-jax-unit (8)
Process completed with exit code 1.
amd64 / test-axlearn-fuji-models-eks
Process completed with exit code 1.
amd64 / test-axlearn-eks
Process completed with exit code 1.
amd64 / test-nccl / nccl-test-gke / nccl-gke (all_gather_perf_mpi)
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-nccl / nccl-test-gke / nccl-gke (reduce_scatter_perf_mpi)
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-nccl / nccl-test-gke / nccl-gke (all_reduce_perf_mpi)
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-nccl / nccl-test-gke / nccl-gke (broadcast_perf_mpi)
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
arm64 / build-jax
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: aws-actions/configure-aws-credentials@v4. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / build-jax
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: aws-actions/configure-aws-credentials@v4. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-te (L0_jax_unittest, 8)
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: actions/checkout@v4, docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-te (L0_jax_distributed_unittest, 8)
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: actions/checkout@v4, docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-cutlass (1, 8)
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: actions/checkout@v4, docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-jax (backend-independent, 1)
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: actions/checkout@v4, docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-maxtext-gke / maxtext-gke (maxtext-cudnn-flash-te-gke, 2, 8, llama2-7b, 1, 1, 2, 4, cudnn_flash_te, minimal_wit...
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-nsys-jax-unit (8)
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: actions/checkout@v4, docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-maxtext-gke / maxtext-gke (maxtext-dot-product-gke, 2, 8, llama2-7b, 1, 1, 2, 4, dot_product, minimal_with_cont...
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-alphafold-inference
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: actions/checkout@v4, docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-alphafold-inference
Unexpected input(s) 'STATISTICS', valid inputs are ['NAME', 'NUM_NODES', 'NUM_GPUS', 'GPU_ARCH', 'MAIN_CONTAINER_NAME', 'JOBSET_TEMPLATE', 'JOBSET_OUTPUT_DIR', 'S3_BUCKET', 'IMAGE', 'COMMAND', 'ENVS', 'EXIT_COMMAND', 'GITHUB_TOKEN', 'NVCR_TOKEN', 'STATISTICS_SCRIPT']
amd64 / test-alphafold-inference
Unexpected input(s) 'STATISTICS', valid inputs are ['NAME', 'NUM_NODES', 'NUM_GPUS', 'GPU_ARCH', 'MAIN_CONTAINER_NAME', 'JOBSET_TEMPLATE', 'JOBSET_OUTPUT_DIR', 'S3_BUCKET', 'IMAGE', 'COMMAND', 'ENVS', 'EXIT_COMMAND', 'GITHUB_TOKEN', 'NVCR_TOKEN', 'STATISTICS_SCRIPT']
amd64 / test-jax (multi-gpu, 8)
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: actions/checkout@v4, docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-jax (single-gpu, 1)
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: actions/checkout@v4, docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-axlearn-fuji-models-eks
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: actions/checkout@v4, docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-maxtext (single-node, 1, 8, 1, 1, 2, 4, NCCL_NET_PLUGIN=none NCCL_NET=Socket, skip_jax_distr...
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: actions/checkout@v4, docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-axlearn-eks
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: actions/checkout@v4, docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-axlearn-eks
Unexpected input(s) 'STATISTICS', valid inputs are ['NAME', 'NUM_NODES', 'NUM_GPUS', 'GPU_ARCH', 'MAIN_CONTAINER_NAME', 'JOBSET_TEMPLATE', 'JOBSET_OUTPUT_DIR', 'S3_BUCKET', 'IMAGE', 'COMMAND', 'ENVS', 'EXIT_COMMAND', 'GITHUB_TOKEN', 'NVCR_TOKEN', 'STATISTICS_SCRIPT']
amd64 / test-axlearn-eks
Unexpected input(s) 'STATISTICS', valid inputs are ['NAME', 'NUM_NODES', 'NUM_GPUS', 'GPU_ARCH', 'MAIN_CONTAINER_NAME', 'JOBSET_TEMPLATE', 'JOBSET_OUTPUT_DIR', 'S3_BUCKET', 'IMAGE', 'COMMAND', 'ENVS', 'EXIT_COMMAND', 'GITHUB_TOKEN', 'NVCR_TOKEN', 'STATISTICS_SCRIPT']
amd64 / test-nsys-jax-nccl (2, 8)
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: actions/checkout@v4, docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
amd64 / test-maxtext (multi-node, 2, 8, 1, 4, 2, 2, OFI_NCCL_PROTOCOL=SENDRECV, jax_distributed_initializ...
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: actions/checkout@v4, docker/login-action@v3. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
publish-containers (maxtext, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:28235461535-maxte... / publish
No files were found with the provided path: final-maxtext.md. No artifacts will be uploaded.
publish-containers (base, mock-jax, 800, ghcr.io/nvidia/jax-toolbox-internal:28235461535-base-amd... / publish
No files were found with the provided path: final-base.md. No artifacts will be uploaded.
publish-containers (axlearn, mock-jax, 900, ghcr.io/nvidia/jax-toolbox-internal:28235461535-axlea... / publish
No files were found with the provided path: final-axlearn.md. No artifacts will be uploaded.
publish-containers (jax, mock-jax, 1000, ghcr.io/nvidia/jax-toolbox-internal:28235461535-jax-amd6... / publish
No files were found with the provided path: final-jax.md. No artifacts will be uploaded.
finalize / upload-badge
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: actions/github-script@v7. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/
finalize / publish-badge
Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: actions/github-script@v7. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/

Artifacts

Produced during runtime
Name Size Digest
NVIDIA~JAX-Toolbox~02QT4D.dockerbuild
66.9 KB
sha256:57521d35f7eae2fb4b398cd15a14f56fa5a21e314c8806f55500785225c0e50b
NVIDIA~JAX-Toolbox~0X9N5Q.dockerbuild
86.7 KB
sha256:6eb944c985eb47a1aef319ae9b5abd8f6c5b2208377f1ff36fc061b62373f2ea
NVIDIA~JAX-Toolbox~25WDOX.dockerbuild
127 KB
sha256:1fd9e6d19fe73888bd1ff535deddde7f3736e051a6df8a40790234d9b8c32efd
NVIDIA~JAX-Toolbox~4542UI.dockerbuild
126 KB
sha256:96977d04c15aa10fd40d0d281f884461baa19a8e4f65e847ba3b49c76764c566
NVIDIA~JAX-Toolbox~4BHKXS.dockerbuild
127 KB
sha256:82e029798be4d1a4ff17837993b3de0790f9e9b29104b39e5077af38b9cbb320
NVIDIA~JAX-Toolbox~6C4UDB.dockerbuild
58.7 KB
sha256:5915ee427f2182125be4f941880281af03e7e835996a74b84f69c77a10ff310b
NVIDIA~JAX-Toolbox~70BPQM.dockerbuild
245 KB
sha256:a8cf3b3ee06927c176b83a24b1c4568db520cb6e0babdd7014d6878cd3533e54
NVIDIA~JAX-Toolbox~84CGM9.dockerbuild
58.2 KB
sha256:5fa64d27261034def9b2376b9d66f6972e8e44f7dee9c8bf8534f8f58fa0ce4f
NVIDIA~JAX-Toolbox~8N8OJ8.dockerbuild
82.4 KB
sha256:a17455ed45a5963897d6288fb6cd20a6c749d63e9e544d3d363158738982bac6
NVIDIA~JAX-Toolbox~CRB1VS.dockerbuild
120 KB
sha256:20d267a23ee346ed8f93c6db9c21cd23e45a333f149d30582d2c096b9d888267
NVIDIA~JAX-Toolbox~E7MGN2.dockerbuild
81 KB
sha256:0e3598a105aaaa095908010552ee1fa0744dd08f8741e0ddb709ebd40b56e674
NVIDIA~JAX-Toolbox~F5NFLF.dockerbuild
117 KB
sha256:283815c3dff43d1906ac7eb144084b514a497927ca978bc549744ef8c3cc3521
NVIDIA~JAX-Toolbox~FR7E4A.dockerbuild
58.2 KB
sha256:bf3dd7c4b43292a48247d193defecb1f66a854caa467dd01f3e238e1a53f2dec
NVIDIA~JAX-Toolbox~J6Y1W8.dockerbuild
110 KB
sha256:602fff778cb0298468a9153baf95ec89b921ed5ad8a6ce9ddccb835f9a3aa8c6
NVIDIA~JAX-Toolbox~J9ZQ76.dockerbuild
128 KB
sha256:0f4a1871d845ec13002b90a221b4bb315e8dedf73736463e1fd8f671be6decbb
NVIDIA~JAX-Toolbox~KFEF6Y.dockerbuild
120 KB
sha256:6a5f1034dc689a99d717cddbe6abb2104beb727242a639e3caaba022bdcd90b0
NVIDIA~JAX-Toolbox~KONMI3.dockerbuild
127 KB
sha256:6db1f5de1c625b322d47d6909087d794032a652103ad4ddeb2fd260e13256037
NVIDIA~JAX-Toolbox~KXL4C6.dockerbuild
64.7 KB
sha256:f8b1b1f0d9f7f2a775305c478e469b660428601f27a61af6e1f150014268fe89
NVIDIA~JAX-Toolbox~LPW3FS.dockerbuild
101 KB
sha256:22f572229db2f1c287d77e3cebb015bb78305fe34b19db5c716ca883fee17dfc
NVIDIA~JAX-Toolbox~NFR36X.dockerbuild
62.9 KB
sha256:b2e40d7f185d64e10b1ab90ef11ad2e0ff4406265fd7fd85ad09d50da9843c8f
NVIDIA~JAX-Toolbox~Q9NR5T.dockerbuild
129 KB
sha256:04e58b2dea2d144795bf0486894286436dad797f174e3f0892c6e87a3c2e582a
NVIDIA~JAX-Toolbox~RUYW2Q.dockerbuild
67.4 KB
sha256:4e2467f595d1d36bf38d2e30b07602914fa54cac85e260de0bae9ab4e1768de5
NVIDIA~JAX-Toolbox~SI6RC6.dockerbuild
81.6 KB
sha256:7f36b2041d01e79b538b6d4466900de70773d82251268cd47601100f173c5964
NVIDIA~JAX-Toolbox~SZ1LLN.dockerbuild
98.8 KB
sha256:55c31dc6e01a5721e3286190d796982f978a8848c5ea9a28963f2a7e3988fd73
NVIDIA~JAX-Toolbox~TFPR0H.dockerbuild
41.6 KB
sha256:d2d22ae849318ebb748eedd55b217084a4b14d24c3048cf09ed4021cc80ee247
NVIDIA~JAX-Toolbox~UESRGA.dockerbuild
118 KB
sha256:cb1b15bb8afb94c270ba7347f0f7e561d015ccdc068c185bb429a867a4336b74
NVIDIA~JAX-Toolbox~UH2IW2.dockerbuild
245 KB
sha256:9a9a75ec9100f620f3b22d44012f38b355dddabd08bb905dee3879d9a571f172
NVIDIA~JAX-Toolbox~WG0KLL.dockerbuild
56.8 KB
sha256:fbadcd7fae53bbbe8bee3cb13f1f3c0904fe008df4d87be119aeaefd339fd893
NVIDIA~JAX-Toolbox~XJ3WQJ.dockerbuild
42.9 KB
sha256:9d304a33bc7a87ff0850b90a296745c7fb3f97b1f5f5a84317e5771af6cadb16
NVIDIA~JAX-Toolbox~XZZUHD.dockerbuild
109 KB
sha256:f9871df7178941a906279645c0f3b92635e9fb8c5407ebe30a72744e3641d528
alphafold-inference
6.12 KB
sha256:ed66fa50b1e74601546ef3293f963127a3dad21bafc26a15a41554f486569d49
alphafold-inference-H100
580 Bytes
sha256:685995dbc1448e24b372ffc0916a88f7f79f938777adcf56099a537576783fd5
artifact-alphafold-build-amd64
574 Bytes
sha256:e20e1b9f920706dc3b1c94de342be3abff1d9d1fdc8497a09eccb5504c93805f
artifact-alphafold-build-arm64
576 Bytes
sha256:5eeface22b671d255c8828412aa2342b5ef5fb3e87c4c6ea5e7a413f8dfb8992
artifact-axlearn-build-amd64
566 Bytes
sha256:63cf841fd5c812165892aca1eb78b595029309157df1284d397eaf6bde4736b0
artifact-axlearn-build-arm64
566 Bytes
sha256:9c400555cf5b36f8dfd10e35396df7f47e43a1ea755aaf21f6e92c434cfe69ab
artifact-base-build-amd64
567 Bytes
sha256:068f42528b677aa35b9d58e25a5e7a3b52fe77f46373011355139cd8be40442c
artifact-base-build-arm64
567 Bytes
sha256:8b741053fc5bf0d603847a20ed4dce42fe4a07aadb740b7e2a16712eb840c6d1
artifact-equinox-build-amd64
568 Bytes
sha256:535e11659e6ce4bacd460a6daf813b6726ef85519f1334be010187f996329dbf
artifact-equinox-build-arm64
568 Bytes
sha256:06fa797c97049ce6423c4f4476dfdf5a8f70ed0f5d69936f9fa443129eaee05d
artifact-final-report
3.45 KB
sha256:e6e1ab4c6f4e32b57b73dbaaadce45fb04efb452d1b8cf684946c50f03ff4dbb
artifact-jax-build-amd64
554 Bytes
sha256:cde241f62fb8fea708d31810dcdb6334493fd05ec21976727b1bcba61de50233
artifact-jax-build-arm64
553 Bytes
sha256:c348aaabd35928b37447fa4fad0ac32b4d0916e2db519693cba01f439fe0454f
artifact-maxtext-build-amd64
568 Bytes
sha256:d4b4fd61b6af05b53db5c5e6ec5d2ef683303c8d24e523cfdd6709d2d39ddfcc
artifact-maxtext-build-arm64
569 Bytes
sha256:71b1b2490941cbfa35b81d74015b730e8c8d4f3bbb6bbaa0c9e1538c9f64b55e
artifact-mpi-operator-compatible-base-build-amd64
637 Bytes
sha256:f09194dc588292a25d841a18345828d69f8788886f13bfbf0af094471578e77c
artifact-nccl-gke-build-amd64
572 Bytes
sha256:9648b7696f02837e02714507545784ae15f4a4306175e08f0c447c1f2a316cb7
artifact-torchax-build-amd64
569 Bytes
sha256:1bc6f3494729286ab0b73ea3dd24233773dacacd1202939596949c462f023abd
artifact-torchax-build-arm64
568 Bytes
sha256:23b581138f71dd5db370ba953e35b5aae38efcebda9061ca93be5b5963a251a8
artifact-workflow-metadata
277 Bytes
sha256:f410bee3db5c5294a1f9fefa06d99a30e5edf049be9f03e7405df3fbfc149f16
axlearn-fuji-3b
19.2 KB
sha256:de9b0b37fb9b7b788c5352e94e78ad76e37329f2fa4f190f101e17d44f910d4e
axlearn-fuji-3b-H100
565 Bytes
sha256:212e2a84b16f62cec962570baf77bb287e1f6e68e2a4738b54104b781ca8d94a
axlearn-unit
394 KB
sha256:e7647957dd8054f80a32c82fe5334b567d051ae1ad18dc28537a5b9d2a3b6003
axlearn-unit-H100
571 Bytes
sha256:8b72d955c8ccbdf4cbed391ea01b3c27ac1e20fe36b36259f0f66ee3b6d2bcf9
bazel-remote-diagnostics-amd64
4.02 MB
sha256:f275eedf6ceb8c3ca69d9b46e23a39bdf3861818475cf0fc8f42ac8a03067663
bazel-remote-diagnostics-arm64
4.11 MB
sha256:c593b4d128afb2089e077f5846fedf4c2a3e3d34f8d6e8f859edd398b47de125
bumped-manifest
5.85 KB
sha256:eed0f633741d84a2dca9a9d49db72b27c6b4eece3d4fe40c02bbb8cf1a103d96
final-equinox
263 Bytes
sha256:396ba807c661f10d08c2b4d9eaf6b5c0897114ebd3e812987513e1e717ea3754
jax-cutlass
13.4 KB
sha256:bc1b1b23b66d4a397209f6f377c04205a99bcbcf59174e1f867237abe5fc8e52
jax-cutlass-H100
556 Bytes
sha256:28a16cddc295fccfe5e5516ab6a9e1bc16e58c4d83cf9807bb474978039c258e
jax-unit-backend-independent
6.96 KB
sha256:09e7fd75192acdd6c59afdd5c9e5efea596a646168dd4257c8f76f8d3f4dc65a
jax-unit-backend-independent-H100
621 Bytes
sha256:b1b76ecd1fd4453f212e48611e86b89838f43d73d553fb28c9222922416b2f47
jax-unit-multi-gpu
10.4 KB
sha256:6c834dfb64b24d8a89ed823e7976fe364bc09da7d1175b05f0627ad4a04b749d
jax-unit-multi-gpu-H100
587 Bytes
sha256:429b929c7daf44dd559da4cda7684b2633cc400e276447f7380892f5d9fe2a79
jax-unit-single-gpu
23 KB
sha256:3ed996cb263bd72414b1aef0df5ed7b2158d1897180eb8ad95323c2c492ca182
jax-unit-single-gpu-H100
592 Bytes
sha256:049fe6b4a698ac48aeb372d060577c1b9940adba72905853d5076fee06bf8beb
maxtext-cudnn-flash-te-gke
62.1 KB
sha256:16d537a5a1549daabf4ea5875ef7c4e348c859e1f3bc6d14f3b8e6113d26b5ae
maxtext-cudnn-flash-te-gke-H100
611 Bytes
sha256:acef3cf86bd5054e2d9e985bd25766760b4952f842eedbef1c7f73ed0480420c
maxtext-dot-product-gke
61.9 KB
sha256:9e4ca534b0085ff25f3b7eafbf068e8beb034b0549bd182f0fbedcc0b8971fec
maxtext-dot-product-gke-H100
597 Bytes
sha256:3ff129c4c0871967ad6ac719d02eeec4929006fcce3854e106a957e3c0da9032
maxtext-multi-node
77.3 KB
sha256:8ff937a0f7876bfdae6c8a31efa7c9817849886904162e574ff6643e3a3d5ce4
maxtext-multi-node-H100
581 Bytes
sha256:82ac41f90b21c34b7400089464290835d824eca1b736e0d18bdb8f726350cdad
maxtext-single-node
48.6 KB
sha256:9d21f0dedb886baaf3d8f6e85d993ff1adcb9d0be21db29af1b5810b65233d9a
maxtext-single-node-H100
582 Bytes
sha256:ddd8059462022d696756179390ac920085fd8ec1e8227e9053db06a1134b7de9
mealkit-axlearn
272 Bytes
sha256:1a307c1bd139803791dc47bd086c26cf26f037d369dfe648476c8f244f3ba16e
mealkit-equinox
272 Bytes
sha256:fd7be6badef3a03ed03c88838916845aab76d578a7f37f6d35b09422bcb7ca9a
mealkit-jax
261 Bytes
sha256:ec20467172d4b1d0ca7f2a1ede695df26d08f97706201ef803c8b8581a37dcb3
mealkit-maxtext
271 Bytes
sha256:c30d4b02721255885ff6ea735a79cf7b02c538fb133e983cf4fbdfe91c3404a3
nccl-gke-all-gather
10.2 KB
sha256:8f0dbc5b33614419e81b030bbc5e3e4c904181c207d197738c3e8ca448e36be0
nccl-gke-all-gather-H100
580 Bytes
sha256:bb2649980db53b493c49a8088567bebfb45534e5193dec7551f79ac18c4bf0fd
nccl-gke-all-reduce
10.4 KB
sha256:c6417b3c80480f36315bb4a57a925ab8227af373f984071fda112fec60ed7aa6
nccl-gke-all-reduce-H100
581 Bytes
sha256:ba9879ec23a2e18a982c1b7ca3bbf32ccb64bb773802d7d141706ec3b81b4892
nccl-gke-broadcast
10.2 KB
sha256:f867ee744696862af22252e13480dbf4c2802981d2ccfecf37cfb386e8814575
nccl-gke-broadcast-H100
577 Bytes
sha256:6d319107d7f41691d71e3a5dd47f4be223ce4b0a005650c0fb1e4150d2fda0ef
nccl-gke-reduce-scatter
10.3 KB
sha256:078a39b5f67a75cc6662565da8518fd8d925dff5149037bbed1326e414c3030c
nccl-gke-reduce-scatter-H100
593 Bytes
sha256:bcbd3ccc58dbd9d7c6a53e30c7a48948d49fcf616a7c95a5b98fa2fdba604daf
nsys-jax-nccl
31.2 MB
sha256:27328c24c2983630b1602812a6185cd2c1840df3b583dfc6f2bdd4d63697b8f8
nsys-jax-nccl-H100
561 Bytes
sha256:2ac99f6f4090c0f631d0f58c8dfe1de21793c833a9cd6d3707d398d3cccfa9d6
nsys-jax-unit
237 MB
sha256:1b1a8a81f1c056fb838c53c27ccb7f1ee77569237d5e952c88a9902f90ec9788
nsys-jax-unit-H100
572 Bytes
sha256:418fc31d2c616853edc6c9f2b19859aa4c931e71222c9a76855a42a4afc8bee3
nsys-jax-version-file
385 Bytes
sha256:b0c9cd1b9868226f37c546eb6f048ae1d7a88b04baf2fee27ac0ecced660feaa
te-L0_jax_distributed_unittest
55.6 KB
sha256:ff3932d27d31fe688edb6785a924d5dbcd28b00a205faf9af85b590814eb6c4c
te-L0_jax_distributed_unittest-H100
628 Bytes
sha256:a3374b2228a20e931c3c54cee225f5326a9278b9454d602f544b2a1b32ac8dc6
te-L0_jax_unittest
1.39 MB
sha256:c660a66f432ce08b4b4d7b8ffb8e1500ab7cf1e6deb2776b79b539d7689b11e4
te-L0_jax_unittest-H100
592 Bytes
sha256:8544713e6073806d7ab9e7d4ba395047a74eb48a2d44c5eb5308d1f9e8cb9e3a