optax.polyak\_sgd
=================

.. currentmodule:: optax

.. autofunction:: polyak_sgd