optax.scale
===========

.. currentmodule:: optax

.. autofunction:: scale