README.md
pyproject.toml
gram_newton_schulz/__init__.py
gram_newton_schulz/autotune_restarts.py
gram_newton_schulz/coefficients.py
gram_newton_schulz/gram_newton_schulz.py
gram_newton_schulz/restart_autotune.py
gram_newton_schulz.egg-info/PKG-INFO
gram_newton_schulz.egg-info/SOURCES.txt
gram_newton_schulz.egg-info/dependency_links.txt
gram_newton_schulz.egg-info/requires.txt
gram_newton_schulz.egg-info/top_level.txt
gram_newton_schulz/muon/__init__.py
gram_newton_schulz/muon/muon.py
gram_newton_schulz/muon/muon_utils/__init__.py
gram_newton_schulz/muon/muon_utils/muon_matrix_split_utils.py
gram_newton_schulz/muon/muon_utils/muon_opt_utils.py