jax:jax is not available for amd64 arch

[mahdikhashan]

hi,

while I am trying to incorporate jax:jax in kubeflow trainer v2.1, I noticed that amd64 architecture is not available.

it would be nice to have this arch.

[olupton]

In general we build for both amd64 and arm64 and only publish if both succeed.
It looks like there was an unusual glitch on the date you cite that led to only one architecture getting published: https://github.com/NVIDIA/JAX-Toolbox/actions/runs/18218585812/job/51873519706
The undated jax tag is just an alias for the most recent jax-YYYY-MM-DD nightly tag.
For example, two days earlier we have both architectures: https://github.com/NVIDIA/JAX-Toolbox/pkgs/container/jax/531757889?tag=jax-2025-10-01

[mahdikhashan]

In general we build for both amd64 and arm64 and only publish if both succeed. It looks like there was an unusual glitch on the date you cite that led to only one architecture getting published: https://github.com/NVIDIA/JAX-Toolbox/actions/runs/18218585812/job/51873519706 The undated jax tag is just an alias for the most recent jax-YYYY-MM-DD nightly tag. For example, two days earlier we have both architectures: https://github.com/NVIDIA/JAX-Toolbox/pkgs/container/jax/531757889?tag=jax-2025-10-01

ok, thanks for clarification.