Add Bazel TPU tests in JAX presubmit.
Examples:
[job with cache hits](https://github.com/jax-ml/jax/actions/runs/16654666670/job/47136442568)
[job without cache hits](https://github.com/jax-ml/jax/actions/runs/16654305268/job/47135173460)
PiperOrigin-RevId: 789790097