quantax.utils.is_sharded_array

quantax.utils.is_sharded_array(array: Array | ndarray | bool | number | bool | int | float | complex) → bool

Whether the input array is sharded across more than one device. Anything that is not a jax.Array (e.g. a numpy array or a Python scalar) is always considered not sharded.