jax.distributed module

Warning

This page was created from a pull request (#9655).

jax.distributed module¶

initialize(coordinator_address, ...)

Initialize distributed system for topology discovery.

previous

jax.dlpack.to_dlpack

next

jax.distributed.initialize