Skip to content

Re-apply Fixed issue with dot_product_attention when using TPU. #21254 after addressing cuDNN/FlashAttention API updates#21333

Merged
fchollet merged 25 commits into
keras-team:masterfrom
pctablet505:master
Jun 10, 2025
Merged

Re-apply Fixed issue with dot_product_attention when using TPU. #21254 after addressing cuDNN/FlashAttention API updates#21333
fchollet merged 25 commits into
keras-team:masterfrom
pctablet505:master

Commits

Commits on May 6, 2025

Commits on May 7, 2025

Commits on May 12, 2025

Commits on May 13, 2025

Commits on May 21, 2025

Commits on May 26, 2025

Commits on May 29, 2025

Commits on Jun 7, 2025

Commits on Jun 10, 2025