quantax.utils.local_to_replicated

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

Assemble identical host-local arrays into a global array replicated on every device (see get_replicated_sharding).

In single-process jobs this is equivalent to to_replicated_array; in multi-host jobs it uses jax.experimental.multihost_utils.host_local_array_to_global_array(). Every process must supply the same local array.