Skip to content

Update nested tensor + MHA tutorial to include SDPA + torch.compile#2813

Merged
jbschlosser merged 3 commits intopytorch:mainfrom
jbschlosser:njt_compile
May 3, 2024
Merged

Update nested tensor + MHA tutorial to include SDPA + torch.compile#2813
jbschlosser merged 3 commits intopytorch:mainfrom
jbschlosser:njt_compile

Commits

Commits on Mar 27, 2024

Commits on Apr 17, 2024

Commits on Apr 30, 2024