jax.debug.visualize\_sharding
=============================

.. currentmodule:: jax.debug

.. autofunction:: visualize_sharding