quantax.utils.to_replicate_numpy

quantax.utils.to_replicate_numpy(array: Array) → ndarray

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