LICENSE
MANIFEST.in
README.md
pyproject.toml
opttx/__init__.py
opttx/apply.py
opttx/curvature.py
opttx/metrics.py
opttx/objective.py
opttx/py.typed
opttx/state.py
opttx/terms.py
opttx.egg-info/PKG-INFO
opttx.egg-info/SOURCES.txt
opttx.egg-info/dependency_links.txt
opttx.egg-info/requires.txt
opttx.egg-info/top_level.txt
opttx/models/__init__.py
opttx/models/gat.py
opttx/models/gcn.py
opttx/optimizers/__init__.py
opttx/optimizers/_utils.py
opttx/optimizers/aa_accelerator.py
opttx/optimizers/adam.py
opttx/optimizers/adamw.py
opttx/optimizers/cg.py
opttx/optimizers/cr.py
opttx/optimizers/lbfgs.py
opttx/optimizers/muon.py
opttx/optimizers/nltgcr.py
opttx/optimizers/nltgcr_crossbatch.py
opttx/optimizers/sgd.py
opttx/optimizers/shampoo.py
opttx/optimizers/soap.py
opttx/optimizers/tgs.py
opttx/optimizers/tgs_accelerator.py
opttx/optimizers/wrapper.py
opttx/solvers/__init__.py
opttx/solvers/cg.py
opttx/solvers/cr.py
opttx/solvers/nltgcr.py
opttx/solvers/tgs.py