optax.ConditionallyTransformState
=================================

.. currentmodule:: optax

.. autoclass:: ConditionallyTransformState