jax

e7fd33e3 - Set PYTHONWARNINGS=error for jax_multiplatform_test.

Set PYTHONWARNINGS=error for jax_multiplatform_test. This is already the default for most of our other test environments, but apparently missed here. PiperOrigin-RevId: 752844454