optax.contrib.muon
==================

.. currentmodule:: optax.contrib

.. autofunction:: muon