Skip to content

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

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

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

Annotations

1 warning

PyTorch Python

succeeded Jan 7, 2025 in 2m 1s