quantax.utils.to_distribute_array

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

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