sharding\_util
==============

.. automodule:: penzai.toolshed.sharding_util

   
   
   .. rubric:: Classes

   .. autosummary::
      :toctree: leaf
      :template: pzclass.rst
   
      ConstrainSharding
      ConstrainShardingByName
   
   

   
   
   .. rubric:: Functions

   .. autosummary::
      :toctree: leaf
      :template: pzbase.rst
   
      name_to_name_device_put
      name_to_name_sharding
      sharded_init