optax.ApplyIfFiniteState
========================

.. currentmodule:: optax

.. autoclass:: ApplyIfFiniteState