jax

dcedfea2 - Guard tests that fail in libtpu==0.0.37, but succeed at nightly libtpu==0.0.38.dev20260315.

Guard tests that fail in libtpu==0.0.37, but succeed at nightly libtpu==0.0.38.dev20260315. Preparing for JAX 0.9.2 release. - 0.0.37: https://github.com/jax-ml/jax/actions/runs/23158234447 - 0.0.38.dev20260315: https://github.com/jax-ml/jax/actions/runs/23162037857 PiperOrigin-RevId: 885189173