optax.PartitionState
====================

.. currentmodule:: optax

.. autoclass:: PartitionState