Skip to content
CI

Dockerfile.jax: install JAX for TE build #3219

Sign in to view logs

Dockerfile.jax: install JAX for TE build

Dockerfile.jax: install JAX for TE build #3219

amd64  /  ...  /  rosetta-pax-multi-node-te (5B_fused_attn_0, 1, 1, 8, 1, 2, --model-type 5B --disable-fused-attn)

amd64 / test-rosetta-pax / rosetta-pax-multi-node-te (5B_fused_attn_0, 1, 1, 8, 1, 2, --model-type 5B --disable-fused-attn)
cancelled Aug 14, 2024 in 1h 53m 46s