LICENSE
README.md
pyproject.toml
heavyball/__init__.py
heavyball/chainable.py
heavyball/utils.py
heavyball.egg-info/PKG-INFO
heavyball.egg-info/SOURCES.txt
heavyball.egg-info/dependency_links.txt
heavyball.egg-info/requires.txt
heavyball.egg-info/top_level.txt
heavyball/optimizations/__init__.py
heavyball/optimizations/integrator.py
heavyball/optimizations/optimizations.py
test/test_bf16_params.py
test/test_bf16_q.py
test/test_bf16_storage.py
test/test_caution.py
test/test_channels_last.py
test/test_closure.py
test/test_ema.py
test/test_foreach.py
test/test_hook.py
test/test_mars.py
test/test_memory.py
test/test_memory_leak.py
test/test_merge.py
test/test_no_grad.py
test/test_soap.py
test/test_stochastic_updates.py
tests/test_psgd_kron_line_optim.py
tests/test_psgd_kron_regression.py
tests/test_psgd_lra_regression.py
tests/test_psgd_optimization.py
tests/test_psgd_optimizations.py
tests/test_psgd_training_performance.py