optax.scale\_by\_rss
====================

.. currentmodule:: optax

.. autofunction:: scale_by_rss