flax.linen.set\_logical\_axis\_rules
====================================

.. currentmodule:: flax.linen

.. autofunction:: set_logical_axis_rules