jax

5b2e5673 - Disable `tests/multiprocess:socket_transfer_test` since it's failing flakily in all backends.