A faster flash attention bwd implementation#177
Open
tonywu95 wants to merge 4 commits intojax-ml:mainfrom
Open
A faster flash attention bwd implementation#177tonywu95 wants to merge 4 commits intojax-ml:mainfrom
tonywu95 wants to merge 4 commits intojax-ml:mainfrom
Commits
Commits on Jun 22, 2023
Commits on Jun 23, 2023
- committed
- committed