Skip to content

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

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

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

Commits

Commits on May 12, 2026

Commits on May 13, 2026