jax

1dbeaf34 - [rollforward]: Add make_transfer_server_interface_factory to the py_socket_transfer jaxlib

[rollforward]: Add make_transfer_server_interface_factory to the py_socket_transfer jaxlib extension so that this can be used as a plugin to xla::ifrt::PjRtClient::CopyArrays. PiperOrigin-RevId: 784315979