Skip to content

[JAX] Support for cuDNN-backed flex attention#2985

Open
vcherepanov-nv wants to merge 13 commits into
NVIDIA:mainfrom
vcherepanov-nv:cudnn-score-mod-jax
Open

[JAX] Support for cuDNN-backed flex attention#2985
vcherepanov-nv wants to merge 13 commits into
NVIDIA:mainfrom
vcherepanov-nv:cudnn-score-mod-jax

Commits

Commits on May 12, 2026

Commits on May 13, 2026

Commits on May 15, 2026

Commits on May 18, 2026

Commits on May 19, 2026

Commits on May 20, 2026

Commits on May 21, 2026

Commits on May 22, 2026