Skip to content

[JAX] Set BSHD as default in Unfused DPA, DPA and MHA API calls#2392

Merged
KshitijLakhani merged 5 commits into
NVIDIA:mainfrom
KshitijLakhani:klakhani/main/set-bshd-default-mha-and-dpa
Nov 21, 2025
Merged

[JAX] Set BSHD as default in Unfused DPA, DPA and MHA API calls#2392
KshitijLakhani merged 5 commits into
NVIDIA:mainfrom
KshitijLakhani:klakhani/main/set-bshd-default-mha-and-dpa

Commits

Commits on Nov 17, 2025

Commits on Nov 21, 2025