quantax.utils.local_to_replicate

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

In multi-host jobs, use jax.experimental.multihost_utils.host_local_array_to_global_array to transform local arrays to be replicated on each device.