optax.contrib.ProdigyState
==========================

.. currentmodule:: optax.contrib

.. autoclass:: ProdigyState