optax.contrib.sophia
====================

.. currentmodule:: optax.contrib

.. autofunction:: sophia