quantax.utils.to_global_array

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

Transform the array to be sharded across all devices in the first dimension. See get_global_sharding for the sharding.