LICENSE
README.md
pyproject.toml
heavyball/__init__.py
heavyball/chainable.py
heavyball/helpers.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_ademamix.py
test/test_bf16_params.py
test/test_bf16_q.py
test/test_bf16_storage.py
test/test_caution.py
test/test_chainable_cpu.py
test/test_channels_last.py
test/test_clip.py
test/test_closure.py
test/test_cpu_features.py
test/test_ema.py
test/test_foreach.py
test/test_helpers_cpu.py
test/test_hook.py
test/test_mars.py
test/test_memory.py
test/test_memory_leak.py
test/test_merge.py
test/test_migrate_cli.py
test/test_nd_param.py
test/test_no_grad.py
test/test_optimizer_cpu_smoke.py
test/test_psgd_precond_init_stability.py
test/test_save_restore.py
test/test_singular_values.py
test/test_soap.py
test/test_stochastic_updates.py
test/test_stochastic_utils_cpu.py
test/test_toy_training.py
test/test_utils_cpu.py
test/test_utils_property.py