quantax.utils.to_replicate_array#
- quantax.utils.to_replicate_array(array: Sequence) Array #
Transform the array to be replicated across all devices. See
get_replicate_sharding
for the sharding.
Transform the array to be replicated across all devices.
See get_replicate_sharding
for the sharding.