Skip to content

Fix backwards attention test for larger head dims#25642

Open
Rifur13 wants to merge 1 commit intojax-ml:mainfrom Rifur13:numerical_stability
+12-10

Commits