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

PyTorch Python

succeeded Jan 7, 2025 in 1m 56s