Skip to content

[JAX] Correct fused attention output after each step of ring attention #6543

[JAX] Correct fused attention output after each step of ring attention

[JAX] Correct fused attention output after each step of ring attention #6543

Annotations

1 warning

JAX C++

succeeded Jan 7, 2025 in 21s