Skip to content

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

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

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

Annotations

1 warning

PyTorch C++

succeeded Jan 7, 2025 in 19s