jax.debug.visualize_array_sharding# jax.debug.visualize_array_sharding(arr, **kwargs)[source]# Visualizes an array’s sharding.
