.. role:: hidden
    :class: hidden-section
.. currentmodule:: brainpy.math.sharding


device_mesh
===========

.. autoclass:: device_mesh
    :members: