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

.. currentmodule:: jax.experimental.shard_map

.. autofunction:: shard_map