Skip to content

Instantly share code, notes, and snippets.

@nvjax
Last active October 6, 2026 09:58
Show Gist options
  • Select an option

  • Save nvjax/913c2af68649fe568e9711c2dabb23ae to your computer and use it in GitHub Desktop.

Select an option

Save nvjax/913c2af68649fe568e9711c2dabb23ae to your computer and use it in GitHub Desktop.
JAX unit test badge
{
"schemaVersion": 1,
"label": "alphafold amd64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "alphafold arm64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "alphafold-inference H100 Unit",
"message": "0/1 passed",
"color": "yellow"
}
{
"schemaVersion": 1,
"label": "axlearn amd64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "axlearn arm64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "axlearn-fuji-3b H100 Unit",
"message": "0/1 passed",
"color": "yellow"
}
{
"schemaVersion": 1,
"label": "axlearn-unit H100 Unit",
"message": "1879/1908 passed",
"color": "yellow"
}
{
"schemaVersion": 1,
"label": "Base image amd64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "Base image arm64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "equinox amd64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "equinox arm64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "jax amd64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "jax arm64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "jax-cutlass H100 Unit",
"message": "5/5 passed",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "jax-unit-backend-independent H100 Unit",
"message": "38/38 passed",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "jax-unit-multi-gpu H100 Unit",
"message": "39/40 passed",
"color": "yellow"
}
{
"schemaVersion": 1,
"label": "jax-unit-single-gpu H100 Unit",
"message": "121/122 passed",
"color": "yellow"
}
{
"schemaVersion": 1,
"label": "jio amd64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "jio arm64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "jio-dec-sync-8b H100 Unit",
"message": "1/1 passed",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "maxtext amd64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "maxtext arm64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "maxtext-cudnn-flash-te-gke H100 Unit",
"message": "0/1 passed",
"color": "yellow"
}
{
"schemaVersion": 1,
"label": "maxtext-dot-product-gke H100 Unit",
"message": "0/1 passed",
"color": "yellow"
}
{
"schemaVersion": 1,
"label": "maxtext-multi-node H100 Unit",
"message": "0/1 passed",
"color": "yellow"
}
{
"schemaVersion": 1,
"label": "maxtext-single-node H100 Unit",
"message": "0/1 passed",
"color": "yellow"
}
{
"schemaVersion": 1,
"label": "mpi-operator-compatible-base amd64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "nccl-gke-all-gather H100 Unit",
"message": "1/1 passed",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "nccl-gke-all-reduce H100 Unit",
"message": "1/1 passed",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "nccl-gke-broadcast H100 Unit",
"message": "1/1 passed",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "nccl-gke amd64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "nccl-gke-reduce-scatter H100 Unit",
"message": "1/1 passed",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "nsys-jax-nccl H100 Unit",
"message": "0/1 passed",
"color": "yellow"
}
{
"schemaVersion": 1,
"label": "nsys-jax-unit H100 Unit",
"message": "105/105 passed",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "te-L0_jax_distributed_unittest H100 Unit",
"message": "122/122 passed",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "te-L0_jax_unittest H100 Unit",
"message": "2616/2617 passed",
"color": "yellow"
}
{
"schemaVersion": 1,
"label": "torchax amd64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "torchax arm64 build",
"message": "pass",
"color": "brightgreen"
}
{
"schemaVersion": 1,
"label": "workflow metadata",
"message": "2026-10-06: run #37444386531",
"color": "bisque"
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment