jax

9ff0b69a - [Pallas/Mosaic GPU] Disable Triton GEMMs in tests that do not use the autotuner.

[Pallas/Mosaic GPU] Disable Triton GEMMs in tests that do not use the autotuner. Using Triton in XLA without the autotuner is not a supported configuration, and running our tests with autotuning is just too slow. PiperOrigin-RevId: 859004996