Skip to content

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

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

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

Annotations

1 warning

PyTorch C++

succeeded Jan 7, 2025 in 20s