3 Commits

Author	SHA1	Message	Date
Ray Smith	2ad82faa1a	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: 852684443	2026-02-13 05:47:03 -08:00
Balazs Racz	baa69dfb78	Makes the entire runtime_config passed into the activations constructor. ... PiperOrigin-RevId: 845153671	2025-12-16 01:56:52 -08:00
Martin Stolle	1014ae9e2a	Adding a simple test for GemmaAttention ... PiperOrigin-RevId: 842135414	2025-12-09 02:13:03 -08:00