flax.linen.with\_logical\_partitioning
======================================

.. currentmodule:: flax.linen

.. autofunction:: with_logical_partitioning