2 Commits

Author	SHA1	Message	Date
Nikhil Dev Goyal	139a8e0964	Replace remaining occurrences of Exp with FastExpMinusOrZero in flash attention. ... PiperOrigin-RevId: 882817324	2026-03-13 08:54:04 -07:00
Ray Smith	76d7951242	Added wheat_from_chaff_test to test the ability of a model to find a needle in a haystack of data. ... Replaced flag with attention_impl to control which attention to run. PiperOrigin-RevId: 869694868	2026-02-13 06:05:30 -08:00