optax.contrib.DoWGState
=======================

.. currentmodule:: optax.contrib

.. autoclass:: DoWGState