jax

a644e23a - [host_callback] Skip test that only works in legacy mode.

[host_callback] Skip test that only works in legacy mode. The jax.experimental.host_callback module is deprecated and will be removed. See https://github.com/google/jax/issues/20385. PiperOrigin-RevId: 680988939