optax.ConditionallyMaskState
============================

.. currentmodule:: optax

.. autoclass:: ConditionallyMaskState