LICENSE
README.md
setup.py
heavyball/__init__.py
heavyball/cached_delayed_psgd_kron.py
heavyball/cached_psgd_kron.py
heavyball/delayed_psgd.py
heavyball/foreach_adamw.py
heavyball/foreach_adopt.py
heavyball/foreach_laprop.py
heavyball/foreach_sfadamw.py
heavyball/foreach_soap.py
heavyball/p_adam.py
heavyball/palm_foreach_sfadamw.py
heavyball/palm_foreach_soap.py
heavyball/precond_schedule_foreach_soap.py
heavyball/precond_schedule_palm_foreach_soap.py
heavyball/precond_schedule_sfpsoap.py
heavyball/psgd_kron.py
heavyball/pure_psgd.py
heavyball/schedule_free_palm_foreach_soap.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
test/test_bf16_params.py
test/test_bf16_q.py
test/test_bf16_storage.py
test/test_caution.py
test/test_closure.py
test/test_ema.py
test/test_foreach.py
test/test_mars.py
test/test_memory.py
test/test_merge.py
test/test_no_grad.py
test/test_psgd.py
test/test_soap.py
test/test_stochastic_updates.py