Reduce the max_examples for splash attention tests

They are too expensive to run all the time. We have 25 test methods, so it's ok
to drop the number of examples each one of them covers.

PiperOrigin-RevId: 707124466

Author: apaszke

tests/pallas/tpu_splash_attention_kernel_test.py

@@ -40,7 +40,7 @@
   raise unittest.SkipTest("these tests require hypothesis")
 
 jax.config.parse_flags_with_absl()
-jtu.setup_hypothesis()
+jtu.setup_hypothesis(max_examples=5)
 
 partial = functools.partial
 Draw = TypeVar("Draw", bound=Callable[[hps.SearchStrategy[Any]], Any])