optax.GradientTransformationExtraArgs
=====================================

.. currentmodule:: optax

.. autoclass:: GradientTransformationExtraArgs