optax.TransformUpdateExtraArgsFn
================================

.. currentmodule:: optax

.. autoclass:: TransformUpdateExtraArgsFn