CHANGELOG.md
LICENSE
MANIFEST.in
README.md
pyproject.toml
v2_additions.md
configs/diloco.yaml
configs/smoke.yaml
configs/v2_smoke.yaml
examples/diloco_master.yaml
examples/train_reference.py
src/speedtronic/__init__.py
src/speedtronic/__main__.py
src/speedtronic/checkpoint.py
src/speedtronic/cli.py
src/speedtronic/config.py
src/speedtronic/data.py
src/speedtronic/hooks.py
src/speedtronic/integrations.py
src/speedtronic/model.py
src/speedtronic/module_utils.py
src/speedtronic/optimizers.py
src/speedtronic/precision.py
src/speedtronic/profiling.py
src/speedtronic/registry.py
src/speedtronic/runtime.py
src/speedtronic/scheduling.py
src/speedtronic/shapes.py
src/speedtronic/trainer.py
src/speedtronic.egg-info/PKG-INFO
src/speedtronic.egg-info/SOURCES.txt
src/speedtronic.egg-info/dependency_links.txt
src/speedtronic.egg-info/entry_points.txt
src/speedtronic.egg-info/requires.txt
src/speedtronic.egg-info/top_level.txt
src/speedtronic/distributed/__init__.py
src/speedtronic/distributed/diloco.py
src/speedtronic/distributed/hub.py
src/speedtronic/distributed/outer.py
src/speedtronic/distributed/tensors.py
tests/test_config.py
tests/test_distributed.py
tests/test_hub.py
tests/test_model.py
tests/test_precision_data.py
tests/test_training.py
tests/test_v2_distributed.py
tests/test_v2_optimizers.py
tests/test_v2_release.py
tests/test_v2_systems.py