optax.scale\_gradient
=====================

.. currentmodule:: optax

.. autofunction:: scale_gradient