Skip to content

[JAX] Add support for sink attention in JAX#2225

Open
pggPL wants to merge 42 commits intoNVIDIA:mainfrom
pggPL:jax_attention
Open

[JAX] Add support for sink attention in JAX#2225
pggPL wants to merge 42 commits intoNVIDIA:mainfrom
pggPL:jax_attention

Commits

Commits on Oct 1, 2025

Commits on Oct 2, 2025

Commits on Oct 6, 2025

Commits on Oct 7, 2025

Commits on Oct 14, 2025

Commits on Oct 15, 2025

Commits on Oct 16, 2025

  • fix

    committed

Commits on Oct 17, 2025

Commits on Oct 21, 2025

Commits on Nov 5, 2025

Commits on Nov 7, 2025