quantax.utils.is_sharded_array

quantax.utils.is_sharded_array(array: Array | ndarray) → bool

Whether the input array is sharded. The array is always considered not sharded if it’s not a jax array.