Skip to content

JAX inference offloading bridge #5037

JAX inference offloading bridge

JAX inference offloading bridge #5037

Triggered via pull request November 14, 2025 22:43
Status Cancelled
Total duration 22m 21s
Artifacts 3

ci.yaml

on: pull_request
metadata
8s
metadata
bump-manifest
20s
bump-manifest
Matrix: amd64 / test-distribution
Matrix: arm64 / test-distribution
amd64  /  ...  /  build-base
2m 40s
amd64 / build-base / build-base
arm64  /  ...  /  build-base
3m 4s
arm64 / build-base / build-base
amd64  /  ...  /  build-mpi-operator-compatible-base
0s
amd64 / test-nccl / build-mpi-operator-compatible-base
amd64  /  ...  /  build-nccl-gke
0s
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-jax-cutlass-h100 / jax-cutlass-test-h100
Waiting for pending jobs
Matrix: amd64 / test-jax / run-unit-test
Waiting for pending jobs
Matrix: amd64 / test-te-a100 / run-unit-test
Waiting for pending jobs
Matrix: amd64 / test-te-h100 / te-test-h100
Waiting for pending jobs
amd64  /  build-maxtext
0s
amd64 / build-maxtext
amd64  /  build-torchax
0s
amd64 / build-torchax
amd64  /  test-nsys-jax-eks
0s
amd64 / test-nsys-jax-eks
amd64  /  ...  /  launch-slurm-runner
amd64 / test-jax / runner / launch-slurm-runner
amd64  /  ...  /  launch-slurm-runner
amd64 / test-te-a100 / runner / launch-slurm-runner
amd64  /  build-upstream-t5x
0s
amd64 / build-upstream-t5x
amd64  /  build-axlearn
0s
amd64 / build-axlearn
Matrix: amd64 / test-nsys-jax / run-unit-test
Waiting for pending jobs
amd64  /  build-equinox
0s
amd64 / build-equinox
amd64  /  ...  /  launch-slurm-runner
amd64 / test-nsys-jax / runner / launch-slurm-runner
Matrix: amd64 / test-nccl / nccl-test
Matrix: amd64 / test-nccl / nccl-test-gke / nccl-gke
Matrix: arm64 / test-jax-cutlass-h100 / jax-cutlass-test-h100
Waiting for pending jobs
Matrix: arm64 / test-jax / run-unit-test
Waiting for pending jobs
Matrix: arm64 / test-te-a100 / run-unit-test
Waiting for pending jobs
Matrix: arm64 / test-te-h100 / te-test-h100
Waiting for pending jobs
arm64  /  build-maxtext
0s
arm64 / build-maxtext
arm64  /  build-torchax
0s
arm64 / build-torchax
arm64  /  test-nsys-jax-eks
0s
arm64 / test-nsys-jax-eks
arm64  /  ...  /  launch-slurm-runner
arm64 / test-jax / runner / launch-slurm-runner
arm64  /  ...  /  launch-slurm-runner
arm64 / test-te-a100 / runner / launch-slurm-runner
arm64  /  build-upstream-t5x
0s
arm64 / build-upstream-t5x
arm64  /  build-axlearn
0s
arm64 / build-axlearn
Matrix: arm64 / test-nsys-jax / run-unit-test
Waiting for pending jobs
arm64  /  build-equinox
0s
arm64 / build-equinox
arm64  /  ...  /  launch-slurm-runner
arm64 / test-nsys-jax / runner / launch-slurm-runner
Matrix: arm64 / test-nccl / nccl-test
Waiting for pending jobs
Matrix: arm64 / test-nccl / nccl-test-gke / nccl-gke
Waiting for pending jobs
amd64  /  ...  /  maxtext-gke-xpk
amd64 / test-maxtext-gke / maxtext-gke-xpk
Matrix: amd64 / test-maxtext / maxtext-multinode
Waiting for pending jobs
Matrix: amd64 / test-maxtext / single-process-multi-device
Waiting for pending jobs
amd64  /  ...  /  build-rosetta
amd64 / build-rosetta-t5x / build-rosetta
amd64  /  test-axlearn-eks
0s
amd64 / test-axlearn-eks
amd64  /  test-axlearn-fuji-models-eks
0s
amd64 / test-axlearn-fuji-models-eks
Matrix: amd64 / test-nsys-jax-archive
arm64  /  ...  /  maxtext-gke-xpk
arm64 / test-maxtext-gke / maxtext-gke-xpk
Matrix: arm64 / test-maxtext / maxtext-multinode
Waiting for pending jobs
Matrix: arm64 / test-maxtext / single-process-multi-device
Waiting for pending jobs
arm64  /  ...  /  build-rosetta
arm64 / build-rosetta-t5x / build-rosetta
arm64  /  test-axlearn-eks
0s
arm64 / test-axlearn-eks
arm64  /  test-axlearn-fuji-models-eks
0s
arm64 / test-axlearn-fuji-models-eks
Matrix: arm64 / test-nsys-jax-archive
amd64  /  ...  /  test-maxtext-metrics
amd64 / test-maxtext / test-maxtext-metrics
amd64  /  collect-docker-tags
0s
amd64 / collect-docker-tags
Matrix: amd64 / test-rosetta-t5x / vit-multi-gpu-multi-node
Waiting for pending jobs
arm64  /  ...  /  test-maxtext-metrics
arm64 / test-maxtext / test-maxtext-metrics
arm64  /  collect-docker-tags
0s
arm64 / collect-docker-tags
Matrix: arm64 / test-rosetta-t5x / vit-multi-gpu-multi-node
Waiting for pending jobs
amd64  /  ...  /  sitrep
amd64 / test-maxtext / test-maxtext-sitrep / sitrep
amd64  /  ...  /  test-t5x-rosetta-summary
amd64 / test-rosetta-t5x / test-t5x-rosetta-summary
amd64  /  ...  /  test-t5x-rosetta-metrics
amd64 / test-rosetta-t5x / test-t5x-rosetta-metrics
arm64  /  ...  /  sitrep
arm64 / test-maxtext / test-maxtext-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
amd64  /  ...  /  test-maxtext-outcome
amd64 / test-maxtext / test-maxtext-outcome
amd64  /  ...  /  sitrep
amd64 / test-rosetta-t5x / test-t5x-rosetta-sitrep / sitrep
arm64  /  ...  /  test-maxtext-outcome
arm64 / test-maxtext / test-maxtext-outcome
arm64  /  ...  /  sitrep
arm64 / test-rosetta-t5x / test-t5x-rosetta-sitrep / sitrep
amd64  /  ...  /  test-t5x-rosetta-outcome
amd64 / test-rosetta-t5x / test-t5x-rosetta-outcome
arm64  /  ...  /  test-t5x-rosetta-outcome
arm64 / test-rosetta-t5x / test-t5x-rosetta-outcome
make-publish-configs
make-publish-configs
merge-new-manifest
merge-new-manifest
Matrix: publish-containers
Waiting for pending jobs
finalize  /  workflow-badge
finalize / workflow-badge
finalize  /  report
finalize / report
finalize  /  upload-badge
finalize / upload-badge
finalize  /  publish-badge
finalize / publish-badge
Fit to window
Zoom out
Zoom in

Annotations

11 errors
amd64 / test-nccl / build-mpi-operator-compatible-base
Canceling since a higher priority waiting request for CI-yhtang/jax-inference-offloading exists
amd64 / test-nccl / nccl-test-gke / build-nccl-gke
Canceling since a higher priority waiting request for CI-yhtang/jax-inference-offloading exists
amd64 / build-jax
Canceling since a higher priority waiting request for CI-yhtang/jax-inference-offloading exists
amd64 / build-jax
The operation was canceled.
arm64 / build-jax
Canceling since a higher priority waiting request for CI-yhtang/jax-inference-offloading exists
arm64 / build-jax
The operation was canceled.
CI
Canceling since a higher priority waiting request for CI-yhtang/jax-inference-offloading exists
CI
Canceling since a higher priority waiting request for CI-yhtang/jax-inference-offloading exists
CI
Canceling since a higher priority waiting request for CI-yhtang/jax-inference-offloading exists
CI
Canceling since a higher priority waiting request for CI-yhtang/jax-inference-offloading exists
CI
Canceling since a higher priority waiting request for CI-yhtang/jax-inference-offloading exists

Artifacts

Produced during runtime
Name Size Digest
artifact-base-build-amd64
567 Bytes
sha256:50acf308c6f4cd628811a82a8bd3e60e29075f70e04f62d893b9c73a26cfc740
artifact-base-build-arm64
566 Bytes
sha256:3c99780510086104ae64905ceffa4602ba34e8bf29ed492a40007b7d8ff26fe9
bumped-manifest
51.6 KB
sha256:4c98b5da1d0a7be70612237eb1bf5c3b97ea8158f5f7b344a9abfbbf4e1a6f4a