``brainpy.math.sharding``: Parallelization Support
==================================================

.. currentmodule:: brainpy.math.sharding
.. automodule:: brainpy.math.sharding


.. autosummary::
   :toctree: generated/
   :nosignatures:
   :template: classtemplate.rst

   device_mesh
   get_sharding
   partition_by_axname
   partition_by_sharding
   partition
   keep_constraint


The commonly used axis names.

.. autosummary::
   :toctree: generated/

   NEU_AXIS
   PRE_AXIS
   POST_AXIS
   SYN_AXIS
   TIME_AXIS
   BATCH_AXIS