jax.distributed module#

initialize([coordinator_address, ...])

Initialize distributed system for topology discovery.