Skip to content

nsys-jax: optimise data loading and .zip creation #3710

nsys-jax: optimise data loading and .zip creation

nsys-jax: optimise data loading and .zip creation #3710

Triggered via pull request January 10, 2025 11:09
Status Failure
Total duration 1d 2h 59m 31s
Artifacts 64

ci.yaml

on: pull_request
metadata
0s
metadata
bump-manifest
33s
bump-manifest
Matrix: amd64 / test-distribution
Matrix: arm64 / test-distribution
amd64  /  ...  /  build-base
4m 7s
amd64 / build-base / build-base
arm64  /  ...  /  build-base
4m 16s
arm64 / build-base / build-base
amd64  /  ...  /  build-jax
9m 4s
amd64 / build-jax / build-jax
arm64  /  ...  /  build-jax
15m 19s
arm64 / build-jax / build-jax
amd64  /  ...  /  build-upstream-pax
9m 37s
amd64 / build-upstream-pax / build-upstream-pax
Matrix: amd64 / test-jax / run-unit-test
amd64  /  ...  /  build-upstream-t5x
13m 28s
amd64 / build-upstream-t5x / build-upstream-t5x
amd64  /  ...  /  launch-slurm-runner
4h 37m
amd64 / test-jax / runner / launch-slurm-runner
amd64  /  test-nsys-jax-eks
5m 8s
amd64 / test-nsys-jax-eks
amd64  /  ...  /  build-gemma
12m 22s
amd64 / build-gemma / build-gemma
amd64  /  ...  /  build-levanter
4m 42s
amd64 / build-levanter / build-levanter
amd64  /  ...  /  build-maxtext
14m 13s
amd64 / build-maxtext / build-maxtext
amd64  /  ...  /  build-triton
20m 2s
amd64 / build-triton / build-triton
Matrix: amd64 / test-nsys-jax / run-unit-test
amd64  /  ...  /  build-equinox
3m 55s
amd64 / build-equinox / build-equinox
amd64  /  ...  /  launch-slurm-runner
4h 30m
amd64 / test-nsys-jax / runner / launch-slurm-runner
arm64  /  ...  /  build-upstream-pax
11m 36s
arm64 / build-upstream-pax / build-upstream-pax
Matrix: arm64 / test-jax / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  build-upstream-t5x
15m 5s
arm64 / build-upstream-t5x / build-upstream-t5x
arm64  /  test-nsys-jax-eks
0s
arm64 / test-nsys-jax-eks
arm64  /  ...  /  launch-slurm-runner
arm64 / test-jax / runner / launch-slurm-runner
arm64  /  ...  /  build
arm64 / build-gemma / build
arm64  /  ...  /  build-levanter
6m 38s
arm64 / build-levanter / build-levanter
arm64  /  ...  /  build-maxtext
15m 16s
arm64 / build-maxtext / build-maxtext
arm64  /  ...  /  build
arm64 / build-triton / build
Matrix: arm64 / test-nsys-jax / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  build-equinox
4m 30s
arm64 / build-equinox / build-equinox
arm64  /  ...  /  launch-slurm-runner
arm64 / test-nsys-jax / runner / launch-slurm-runner
amd64  /  ...  /  build-rosetta
amd64 / build-rosetta-pax / build-rosetta
Matrix: amd64 / test-te / run-unit-test
Waiting for pending jobs
amd64  /  ...  /  launch-slurm-runner
amd64 / test-te / runner / launch-slurm-runner
Matrix: amd64 / test-upstream-pax / pax-multi-node
Waiting for pending jobs
Matrix: amd64 / test-upstream-pax / single-process-evaluation
Waiting for pending jobs
Matrix: amd64 / test-upstream-pax / single-process-multi-device
Waiting for pending jobs
Matrix: amd64 / test-te-multigpu / te-multi-gpu
Waiting for pending jobs
amd64  /  ...  /  build-rosetta
12m 51s
amd64 / build-rosetta-t5x / build-rosetta
Matrix: amd64 / test-upstream-t5x / t5x-multi-gpu
Matrix: amd64 / test-upstream-t5x / t5x-multi-node
Matrix: amd64 / test-gemma / run-unit-test
amd64  /  ...  /  launch-slurm-runner
3h 25m
amd64 / test-gemma / runner / launch-slurm-runner
Matrix: amd64 / test-levanter / run-unit-test
amd64  /  ...  /  launch-slurm-runner
3h 22m
amd64 / test-levanter / runner / launch-slurm-runner
Matrix: amd64 / test-maxtext / maxtext-multinode
Matrix: amd64 / test-maxtext / single-process-multi-device
Matrix: amd64 / test-triton / run-unit-test
amd64  /  ...  /  launch-slurm-runner
3h 47m
amd64 / test-triton / runner / launch-slurm-runner
Matrix: amd64 / test-nsys-jax-archive
arm64  /  ...  /  build-rosetta
arm64 / build-rosetta-pax / build-rosetta
Matrix: arm64 / test-te / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  launch-slurm-runner
arm64 / test-te / runner / launch-slurm-runner
Matrix: arm64 / test-upstream-pax / pax-multi-node
Waiting for pending jobs
Matrix: arm64 / test-upstream-pax / single-process-evaluation
Waiting for pending jobs
Matrix: arm64 / test-upstream-pax / single-process-multi-device
Waiting for pending jobs
Matrix: arm64 / test-te-multigpu / te-multi-gpu
Waiting for pending jobs
arm64  /  ...  /  build-rosetta
15m 57s
arm64 / build-rosetta-t5x / build-rosetta
Matrix: arm64 / test-upstream-t5x / t5x-multi-gpu
Waiting for pending jobs
Matrix: arm64 / test-upstream-t5x / t5x-multi-node
Waiting for pending jobs
Matrix: arm64 / test-gemma / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  launch-slurm-runner
arm64 / test-gemma / runner / launch-slurm-runner
Matrix: arm64 / test-levanter / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  launch-slurm-runner
arm64 / test-levanter / runner / launch-slurm-runner
Matrix: arm64 / test-maxtext / maxtext-multinode
Waiting for pending jobs
Matrix: arm64 / test-maxtext / single-process-multi-device
Waiting for pending jobs
Matrix: arm64 / test-triton / run-unit-test
Waiting for pending jobs
arm64  /  ...  /  launch-slurm-runner
arm64 / test-triton / runner / launch-slurm-runner
Matrix: arm64 / test-nsys-jax-archive
Matrix: amd64 / test-rosetta-pax / rosetta-pax-multi-node-te
Waiting for pending jobs
Matrix: amd64 / test-rosetta-pax / rosetta-pax-multi-node
Waiting for pending jobs
Matrix: amd64 / test-rosetta-pax / rosetta-pax-single-node-dropout-te
Waiting for pending jobs
Matrix: amd64 / test-rosetta-pax / single-process-evaluation-te
Waiting for pending jobs
Matrix: amd64 / test-rosetta-pax / single-process-multi-device-te
Waiting for pending jobs
amd64  /  ...  /  test-upstream-pax-summary
amd64 / test-upstream-pax / test-upstream-pax-summary
amd64  /  ...  /  test-upstream-pax-metrics
amd64 / test-upstream-pax / test-upstream-pax-metrics
amd64  /  ...  /  sitrep
amd64 / test-te-multigpu / sitrep
amd64  /  collect-docker-tags
0s
amd64 / collect-docker-tags
Matrix: amd64 / test-rosetta-t5x / multi-gpu-multi-node
Matrix: amd64 / test-rosetta-t5x / single-process-multi-device
Matrix: amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node
Matrix: amd64 / test-rosetta-t5x / vit-single-process-multi-device
amd64  /  ...  /  test-upstream-t5x-summary
0s
amd64 / test-upstream-t5x / test-upstream-t5x-summary
amd64  /  ...  /  test-upstream-t5x-metrics
0s
amd64 / test-upstream-t5x / test-upstream-t5x-metrics
amd64  /  ...  /  test-maxtext-summary
0s
amd64 / test-maxtext / test-maxtext-summary
amd64  /  ...  /  test-maxtext-metrics
0s
amd64 / test-maxtext / test-maxtext-metrics
Matrix: arm64 / test-rosetta-pax / rosetta-pax-multi-node-te
Waiting for pending jobs
Matrix: arm64 / test-rosetta-pax / rosetta-pax-multi-node
Waiting for pending jobs
Matrix: arm64 / test-rosetta-pax / rosetta-pax-single-node-dropout-te
Waiting for pending jobs
Matrix: arm64 / test-rosetta-pax / single-process-evaluation-te
Waiting for pending jobs
Matrix: arm64 / test-rosetta-pax / single-process-multi-device-te
Waiting for pending jobs
arm64  /  ...  /  test-upstream-pax-summary
arm64 / test-upstream-pax / test-upstream-pax-summary
arm64  /  ...  /  test-upstream-pax-metrics
arm64 / test-upstream-pax / test-upstream-pax-metrics
arm64  /  ...  /  sitrep
arm64 / test-te-multigpu / sitrep
arm64  /  collect-docker-tags
0s
arm64 / collect-docker-tags
Matrix: arm64 / test-rosetta-t5x / multi-gpu-multi-node
Waiting for pending jobs
Matrix: arm64 / test-rosetta-t5x / single-process-multi-device
Waiting for pending jobs
Matrix: arm64 / test-rosetta-t5x / vit-multi-gpu-multi-node
Waiting for pending jobs
Matrix: arm64 / test-rosetta-t5x / vit-single-process-multi-device
Waiting for pending jobs
arm64  /  ...  /  test-upstream-t5x-summary
arm64 / test-upstream-t5x / test-upstream-t5x-summary
arm64  /  ...  /  test-upstream-t5x-metrics
arm64 / test-upstream-t5x / test-upstream-t5x-metrics
arm64  /  ...  /  test-maxtext-summary
arm64 / test-maxtext / test-maxtext-summary
arm64  /  ...  /  test-maxtext-metrics
arm64 / test-maxtext / test-maxtext-metrics
amd64  /  ...  /  test-pax-rosetta-summary
amd64 / test-rosetta-pax / test-pax-rosetta-summary
amd64  /  ...  /  test-pax-rosetta-metrics
amd64 / test-rosetta-pax / test-pax-rosetta-metrics
amd64  /  ...  /  sitrep
amd64 / test-upstream-pax / test-upstream-pax-sitrep / sitrep
amd64  /  ...  /  test-t5x-rosetta-summary
0s
amd64 / test-rosetta-t5x / test-t5x-rosetta-summary
amd64  /  ...  /  test-t5x-rosetta-metrics
0s
amd64 / test-rosetta-t5x / test-t5x-rosetta-metrics
amd64  /  ...  /  sitrep
12s
amd64 / test-upstream-t5x / test-upstream-t5x-sitrep / sitrep
amd64  /  ...  /  sitrep
6s
amd64 / test-maxtext / test-maxtext-sitrep / sitrep
arm64  /  ...  /  test-pax-rosetta-summary
arm64 / test-rosetta-pax / test-pax-rosetta-summary
arm64  /  ...  /  test-pax-rosetta-metrics
arm64 / test-rosetta-pax / test-pax-rosetta-metrics
arm64  /  ...  /  sitrep
arm64 / test-upstream-pax / test-upstream-pax-sitrep / sitrep
arm64  /  ...  /  test-t5x-rosetta-summary
arm64 / test-rosetta-t5x / test-t5x-rosetta-summary
arm64  /  ...  /  test-t5x-rosetta-metrics
arm64 / test-rosetta-t5x / test-t5x-rosetta-metrics
arm64  /  ...  /  sitrep
arm64 / test-upstream-t5x / test-upstream-t5x-sitrep / sitrep
arm64  /  ...  /  sitrep
arm64 / test-maxtext / test-maxtext-sitrep / sitrep
amd64  /  ...  /  sitrep
amd64 / test-rosetta-pax / test-pax-rosetta-sitrep / sitrep
amd64  /  ...  /  test-upstream-pax-outcome
amd64 / test-upstream-pax / test-upstream-pax-outcome
amd64  /  ...  /  sitrep
6s
amd64 / test-rosetta-t5x / test-t5x-rosetta-sitrep / sitrep
amd64  /  ...  /  test-upstream-t5x-outcome
0s
amd64 / test-upstream-t5x / test-upstream-t5x-outcome
amd64  /  ...  /  test-maxtext-outcome
0s
amd64 / test-maxtext / test-maxtext-outcome
arm64  /  ...  /  sitrep
arm64 / test-rosetta-pax / test-pax-rosetta-sitrep / sitrep
arm64  /  ...  /  test-upstream-pax-outcome
arm64 / test-upstream-pax / test-upstream-pax-outcome
arm64  /  ...  /  sitrep
arm64 / test-rosetta-t5x / test-t5x-rosetta-sitrep / sitrep
arm64  /  ...  /  test-upstream-t5x-outcome
arm64 / test-upstream-t5x / test-upstream-t5x-outcome
arm64  /  ...  /  test-maxtext-outcome
arm64 / test-maxtext / test-maxtext-outcome
amd64  /  ...  /  test-pax-rosetta-outcome
amd64 / test-rosetta-pax / test-pax-rosetta-outcome
amd64  /  ...  /  test-t5x-rosetta-outcome
0s
amd64 / test-rosetta-t5x / test-t5x-rosetta-outcome
arm64  /  ...  /  test-pax-rosetta-outcome
arm64 / test-rosetta-pax / test-pax-rosetta-outcome
arm64  /  ...  /  test-t5x-rosetta-outcome
arm64 / test-rosetta-t5x / test-t5x-rosetta-outcome
make-publish-configs
3s
make-publish-configs
merge-new-manifest
0s
merge-new-manifest
Matrix: publish-containers
finalize  /  workflow-badge
4s
finalize / workflow-badge
finalize  /  report
14s
finalize / report
finalize  /  upload-badge
5s
finalize / upload-badge
finalize  /  publish-badge
2s
finalize / publish-badge
Fit to window
Zoom out
Zoom in

Annotations

21 errors
arm64 / build-upstream-pax / build-upstream-pax
buildx failed with: ERROR: failed to solve: process "/bin/sh -c <<\"EOF\" bash -exu -o pipefail\ngit-clone.sh ${URLREF_LINGVO} ${SRC_PATH_LINGVO}\npushd ${SRC_PATH_LINGVO}\n\nCPU_ARCH=\"$(dpkg --print-architecture)\"\nif [[ \"${CPU_ARCH}\" == \"arm64\" ]]; then\n\n# Use aarch distribution of protobufs\npatch -p1 <<\"EOFINNER\"\ndiff --git a/lingvo/repo.bzl b/lingvo/repo.bzl\nindex ce65822d2..d9c0277aa 100644\n--- a/lingvo/repo.bzl\n+++ b/lingvo/repo.bzl\n@@ -232,9 +232,9 @@ filegroup(\n )\n \"\"\",\n urls = [\n- \"https://github.com/protocolbuffers/protobuf/releases/download/v21.9/protoc-21.9-linux-x86_64.zip\",\n+ \"https://github.com/protocolbuffers/protobuf/releases/download/v21.9/protoc-21.9-linux-aarch_64.zip\",\n ],\n- sha256 = \"3cd951aff8ce713b94cde55e12378f505f2b89d47bf080508cf77e3934f680b6\",\n+ sha256 = \"a584286dfa8ebb17032ece206ed74d5e9931e2edb9016e427be2a0dab3b21071\",\n )\n\n def icu():\nEOFINNER\n\nfi\n\npip install tensorflow_datasets==4.9.2 auditwheel tensorflow==2.18.0 /opt/tensorflow_text*.whl\nfor pattern in \\\n \"s|tensorflow=|#tensorflow=|g\" \\\n \"s|tensorflow-text=|#tensorflow-text=|g\" \\\n \"s|dataclasses=|#dataclasses=|g\" \\\n \"s|==.*||g\" \\\n; do\n sed -i \"${pattern}\" ${SRC_PATH_LINGVO}/docker/dev.requirements.txt\ndone\n# Lingvo support only python < 3.12, so we hack it and update dependencies\n# to be able to build for py-3.12\nfor pattern in \\\n \"s|tensorflow-text~=2.13.0|tensorflow-text~=2.18.0|g\" \\\n \"s|tensorflow~=2.13.0|tensorflow~=2.18.0|g\" \\\n \"s|python_requires='>=3.8,<3.11'|python_requires='>=3.8,<3.13'|\" \\\n; do\n sed -i \"${pattern}\" ${SRC_PATH_LINGVO}/pip_package/setup.py;\ndone\npip install -r docker/dev.requirements.txt\n\n# Some tests are flaky right now, so we skip running the tests.\nBUILD_ARCH=\"x86_64\"\nif [[ \"$CPU_ARCH\" == \"arm64\" ]]; then\n BUILD_ARCH=\"aarch64\";\nfi\nsed -i 's/manylinux2014_x86_64/manylinux_2_38_'\"${BUILD_ARCH}\"'/' pip_package/build.sh\nSKIP_TESTS=1 PYTHON_MINOR_VERSION=$(python --version | cut -d ' ' -f 2 | cut -d '.' -f 2) pip_package/build.sh\nEOF" did not complete successfully: exit code: 1
amd64 / build-upstream-pax / build-upstream-pax
buildx failed with: ERROR: failed to solve: process "/bin/sh -c <<\"EOF\" bash -exu -o pipefail\ngit-clone.sh ${URLREF_LINGVO} ${SRC_PATH_LINGVO}\npushd ${SRC_PATH_LINGVO}\n\nCPU_ARCH=\"$(dpkg --print-architecture)\"\nif [[ \"${CPU_ARCH}\" == \"arm64\" ]]; then\n\n# Use aarch distribution of protobufs\npatch -p1 <<\"EOFINNER\"\ndiff --git a/lingvo/repo.bzl b/lingvo/repo.bzl\nindex ce65822d2..d9c0277aa 100644\n--- a/lingvo/repo.bzl\n+++ b/lingvo/repo.bzl\n@@ -232,9 +232,9 @@ filegroup(\n )\n \"\"\",\n urls = [\n- \"https://github.com/protocolbuffers/protobuf/releases/download/v21.9/protoc-21.9-linux-x86_64.zip\",\n+ \"https://github.com/protocolbuffers/protobuf/releases/download/v21.9/protoc-21.9-linux-aarch_64.zip\",\n ],\n- sha256 = \"3cd951aff8ce713b94cde55e12378f505f2b89d47bf080508cf77e3934f680b6\",\n+ sha256 = \"a584286dfa8ebb17032ece206ed74d5e9931e2edb9016e427be2a0dab3b21071\",\n )\n\n def icu():\nEOFINNER\n\nfi\n\npip install tensorflow_datasets==4.9.2 auditwheel tensorflow==2.18.0 /opt/tensorflow_text*.whl\nfor pattern in \\\n \"s|tensorflow=|#tensorflow=|g\" \\\n \"s|tensorflow-text=|#tensorflow-text=|g\" \\\n \"s|dataclasses=|#dataclasses=|g\" \\\n \"s|==.*||g\" \\\n; do\n sed -i \"${pattern}\" ${SRC_PATH_LINGVO}/docker/dev.requirements.txt\ndone\n# Lingvo support only python < 3.12, so we hack it and update dependencies\n# to be able to build for py-3.12\nfor pattern in \\\n \"s|tensorflow-text~=2.13.0|tensorflow-text~=2.18.0|g\" \\\n \"s|tensorflow~=2.13.0|tensorflow~=2.18.0|g\" \\\n \"s|python_requires='>=3.8,<3.11'|python_requires='>=3.8,<3.13'|\" \\\n; do\n sed -i \"${pattern}\" ${SRC_PATH_LINGVO}/pip_package/setup.py;\ndone\npip install -r docker/dev.requirements.txt\n\n# Some tests are flaky right now, so we skip running the tests.\nBUILD_ARCH=\"x86_64\"\nif [[ \"$CPU_ARCH\" == \"arm64\" ]]; then\n BUILD_ARCH=\"aarch64\";\nfi\nsed -i 's/manylinux2014_x86_64/manylinux_2_38_'\"${BUILD_ARCH}\"'/' pip_package/build.sh\nSKIP_TESTS=1 PYTHON_MINOR_VERSION=$(python --version | cut -d ' ' -f 2 | cut -d '.' -f 2) pip_package/build.sh\nEOF" did not complete successfully: exit code: 1
amd64 / test-triton / triton-V100-unit-test
Process completed with exit code 1.
amd64 / test-triton / triton-A100-unit-test
Process completed with exit code 1.
amd64 / test-upstream-t5x / t5x-multi-node (8G2N, 8, 2)
The job running on runner jumpbox-vc69x-b8f26 has exceeded the maximum execution time of 360 minutes.
amd64 / test-upstream-t5x / t5x-multi-node (8G2N, 8, 2)
The operation was canceled.
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N8G-te-1, 8, 2, --gin.train/utils.DatasetConfig.pack=False --gin.train_eva...
The job running on runner jumpbox-vc69x-fcb56 has exceeded the maximum execution time of 360 minutes.
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 2)
The job running on runner jumpbox-vc69x-mvvzf has exceeded the maximum execution time of 360 minutes.
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (1, 2)
The operation was canceled.
amd64 / test-upstream-t5x / t5x-multi-node (8G2N_fmha, 8, 2, --enable-fmha 1)
The job running on runner jumpbox-vc69x-rdt2z has exceeded the maximum execution time of 360 minutes.
amd64 / test-upstream-t5x / test-upstream-t5x-outcome
Process completed with exit code 1.
amd64 / test-rosetta-t5x / multi-gpu-multi-node (2N2G_te-0, 2, 2, --enable-te 0)
The job running on runner jumpbox-vc69x-plhvr has exceeded the maximum execution time of 360 minutes.
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 2)
The job running on runner jumpbox-vc69x-khsjf has exceeded the maximum execution time of 360 minutes.
amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node (8, 2)
The operation was canceled.
amd64 / test-rosetta-t5x / test-t5x-rosetta-outcome
Process completed with exit code 1.
amd64 / test-maxtext / maxtext-multinode (1, 4, 2, 2)
The job running on runner jumpbox-vc69x-m4dxp has exceeded the maximum execution time of 360 minutes.
amd64 / test-maxtext / maxtext-multinode (1, 4, 2, 2)
The operation was canceled.
amd64 / test-maxtext / test-maxtext-outcome
Process completed with exit code 1.

Artifacts

Produced during runtime
Name Size
artifact-base-build-amd64
566 Bytes
artifact-base-build-arm64
567 Bytes
artifact-equinox-build-amd64
569 Bytes
artifact-equinox-build-arm64
569 Bytes
artifact-final-report
3.78 KB
artifact-gemma-build-amd64
560 Bytes
artifact-jax-build-amd64
553 Bytes
artifact-jax-build-arm64
552 Bytes
artifact-levanter-build-amd64
572 Bytes
artifact-levanter-build-arm64
572 Bytes
artifact-maxtext-build-amd64
568 Bytes
artifact-maxtext-build-arm64
568 Bytes
artifact-maxtext-test
654 Bytes
artifact-pax-build-amd64
472 Bytes
artifact-pax-build-arm64
471 Bytes
artifact-rosetta-build-t5x-amd64
584 Bytes
artifact-rosetta-build-t5x-arm64
584 Bytes
artifact-rosetta-t5x-mgmn-test
658 Bytes
artifact-t5x-build-amd64
568 Bytes
artifact-t5x-build-arm64
567 Bytes
artifact-triton-build-amd64
565 Bytes
artifact-upstream-t5x-mgmn-test
634 Bytes
artifact-workflow-metadata
277 Bytes
bumped-manifest
81.2 KB
final-base
254 Bytes
final-equinox
263 Bytes
final-gemma
257 Bytes
final-jax
251 Bytes
final-levanter
266 Bytes
final-maxtext
263 Bytes
final-t5x
251 Bytes
final-triton
260 Bytes
final-upstream-t5x
277 Bytes
gemma-unit-test-A100
5.25 KB
gemma-unit-test-V100
4.03 KB
jax-unit-test-A100
20.2 KB
jax-unit-test-V100
26 KB
levanter-unit-test-A100
7.36 KB
levanter-unit-test-V100
9.05 KB
mealkit-equinox
272 Bytes
mealkit-gemma
266 Bytes
mealkit-jax
261 Bytes
mealkit-levanter
275 Bytes
mealkit-maxtext
272 Bytes
mealkit-t5x
261 Bytes
mealkit-triton
269 Bytes
mealkit-upstream-t5x
287 Bytes
nsys-jax-unit-test-A100
32.5 MB
nsys-jax-unit-test-V100
17.4 MB
rosetta-t5x-12708375325-1P1G_te-0
35.2 KB
rosetta-t5x-12708375325-1P8G_te-1
35.3 KB
rosetta-t5x-vit-12708375325-VIT1G1N
21.7 KB
rosetta-t5x-vit-12708375325-VIT1P8G
21.9 KB
rosetta-t5x-vit-12708375325-VIT8G1N
33.7 KB
triton-unit-test-A100
3.09 KB
triton-unit-test-V100
3.08 KB
upstream-maxtext-12708375325-1DP1FSDP1TP1PP
15.9 KB
upstream-maxtext-12708375325-1DP1FSDP8TP1PP
22.5 KB
upstream-maxtext-12708375325-1DP2FSDP4TP1PP_single_process
16.3 KB
upstream-maxtext-12708375325-1DP4FSDP2TP1PP
22.1 KB
upstream-maxtext-12708375325-1DP8FSDP1TP1PP
22.2 KB
upstream-maxtext-12708375325-2DP2FSDP2TP1PP
22.5 KB
upstream-t5x-12708375325-1P2G_fmha
34 KB
upstream-t5x-12708375325-1P8G
33.6 KB