Add init_system API for multi-host GPU.#8364

Merged

copybara-service[bot] merged 1 commit intojax-ml:mainfrom zhangqiaorjc:dsys

Oct 28, 2021

+85-5

Commits on Oct 26, 2021

Add jax.distributed.initialize for multi-host GPU.
zhangqiaorjc
committed