optax.MaskedState
=================

.. currentmodule:: optax

.. autoclass:: MaskedState