Parameter Groups for Optimizers
===============================


.. autofunction:: nequip.model.MuonParamGroups