.gitignore
LICENSE
Makefile
README.md
conf.py
doc-requirements.txt
index.rst
make.bat
mypy.ini
pyproject.toml
setup.py
.github/workflows/black.yml
.github/workflows/mypy.yml
.github/workflows/pypi-publish.yml
bagua/__init__.py
bagua/bagua_define.py
bagua/version.py
bagua.egg-info/PKG-INFO
bagua.egg-info/SOURCES.txt
bagua.egg-info/dependency_links.txt
bagua.egg-info/entry_points.txt
bagua.egg-info/requires.txt
bagua.egg-info/top_level.txt
bagua/autotune/__init__.py
bagua/distributed/__init__.py
bagua/distributed/launch.py
bagua/script/__init__.py
bagua/script/bagua_sys_perf
bagua/script/baguarun.py
bagua/service/__init__.py
bagua/service/autotune_service.py
bagua/service/service_discovery.py
bagua/torch_api/__init__.py
bagua/torch_api/communication.py
bagua/torch_api/compression.py
bagua/torch_api/distributed.py
bagua/torch_api/distributed_define.py
bagua/torch_api/env.py
bagua/torch_api/exceptions.py
bagua/torch_api/fuse_optimizer.py
bagua/torch_api/utils.py
bagua/torch_api/algorithms/__init__.py
bagua/torch_api/algorithms/distributed_module.py
bagua/torch_api/algorithms/allreduce/__init__.py
bagua/torch_api/algorithms/allreduce/allreducer.py
bagua/torch_api/algorithms/allreduce/sg_allreducer.py
bagua/torch_api/algorithms/decentralize/__init__.py
bagua/torch_api/algorithms/decentralize/decentralized_reducer.py
bagua/torch_api/contrib/__init__.py
bagua/torch_api/contrib/data/__init__.py
bagua/torch_api/contrib/data/load_balancing_data_loader.py
baguaelastic/distributed/launch.py
docs/.gitignore
docs/index.md
docs/templates/_lunr_search.inc.mako
docs/templates/config.mako
docs/templates/credits.mako
docs/templates/css.mako
docs/templates/head.mako
docs/templates/html.mako
docs/templates/logo.mako
docs/templates/pdf.mako
docs/templates/search.mako
docs/templates/text.mako
tests/__init__.py
tests/test_communication.py
tests/test_flatten.py
tests/test_fused_optimizer.py
tests/autotune/test_bayesian_optimizer.py
tests/nccl/__init__.py
tests/nccl/test_gather.py
tests/nccl/test_p2p.py
tests/qsgd/test_norm_qsgd.py
tests/qsgd/test_qsgd.py
tests/qsgd/utils.py
tests/service/test_autotune_service.py
tests/service/test_service_discovery.py
tests/tools/test_loadbalance.py