Skip to content

[JAX] [PyT] [Common] Enable D=256 BWD cuDNN fused attn for Blackwell CC 10.x #3056

Open
KshitijLakhani wants to merge 26 commits into
NVIDIA:mainfrom
KshitijLakhani:klakhani/feat/enable-headdim256-bwd-sm100
Open

[JAX] [PyT] [Common] Enable D=256 BWD cuDNN fused attn for Blackwell CC 10.x #3056
KshitijLakhani wants to merge 26 commits into
NVIDIA:mainfrom
KshitijLakhani:klakhani/feat/enable-headdim256-bwd-sm100

[pre-commit.ci] auto fixes from pre-commit.com hooks

6f2b748
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Jul 23, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO