optax.partition
===============

.. currentmodule:: optax

.. autofunction:: partition