quantax.utils.local_to_global

quantax.utils.local_to_global(array: Array | ndarray | bool | number | bool | int | float | complex) → Array

Assemble the host-local arrays of all processes into a single global array sharded along its first dimension (see get_distributed_sharding).

In single-process jobs this is equivalent to to_distributed_array; in multi-host jobs it uses jax.experimental.multihost_utils.host_local_array_to_global_array().