quantax.utils.local_to_global

quantax.utils.local_to_global(array: Sequence) → Array

In multi-host jobs, use jax.experimental.multihost_utils.host_local_array_to_global_array to transform local arrays to be sharded.