Skip to content

[JAX][Common] Enable cuDNN fused attn backend for NO_MASK + bidirectional SWA#2961

Merged
KshitijLakhani merged 6 commits into
NVIDIA:mainfrom
KshitijLakhani:klakhani/fix/no-mask-swa-attn
May 6, 2026
Merged

[JAX][Common] Enable cuDNN fused attn backend for NO_MASK + bidirectional SWA#2961
KshitijLakhani merged 6 commits into
NVIDIA:mainfrom
KshitijLakhani:klakhani/fix/no-mask-swa-attn

Commits

Commits on May 6, 2026