optax.contrib.MuonState
=======================

.. currentmodule:: optax.contrib

.. autoclass:: MuonState