jax.shard\_map
==============

.. currentmodule:: jax

.. autofunction:: shard_map