Skip to content
Back to pull request #22361

Fix #20429 — Attention layer hardcodes 3-D shape assumptions, breaking N-D inputs #10757

Sign in to view logs

Fix #20429 — Attention layer hardcodes 3-D shape assumptions, breaking N-D inputs

Fix #20429 — Attention layer hardcodes 3-D shape assumptions, breaking N-D inputs #10757

Run tests (3.11, jax, nnx_enabled = true)

Run tests (3.11, jax, nnx_enabled = true)
succeeded Mar 5, 2026 in 3m 26s