brainpy.math.sharding.BATCH\_AXIS
=================================

.. currentmodule:: brainpy.math.sharding

.. autodata:: BATCH_AXIS