optax.TransformUpdateFn
=======================

.. currentmodule:: optax

.. autoclass:: TransformUpdateFn