quantax.utils.global_to_local

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

Convert a distributed global array into the host-local array holding only this process’s shards, using jax.experimental.multihost_utils.global_array_to_host_local_array().

In single-process jobs the array is already local and is returned unchanged.