jax

ab90f913 - Add Bazel TPU tests in JAX presubmit.

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