brainstate.transform.shard\_map
===============================

.. currentmodule:: brainstate.transform

.. autofunction:: shard_map