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

Commit
197 days ago
[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
Author
Parents
Loading