quantax.utils.global_to_local

quantax.utils.global_to_local(array: Array) → Array

In multi-host jobs, use jax.experimental.multihost_utils.global_array_to_host_local_array to transform a sharded array to be local on each device.