optax.GradientTransformation
============================

.. currentmodule:: optax

.. autoclass:: GradientTransformation