.gitignore
CHANGELOG.md
LICENSE
Makefile
README.md
install.sh
pyproject.toml
setup.cfg
setup.py
.github/dependabot.yml
.github/issue-close-app.yml
.github/ISSUE_TEMPLATE/bug_report.md
.github/ISSUE_TEMPLATE/feature_request.md
.github/workflows/black.yml
.github/workflows/codeql-analysis.yml
.github/workflows/flake8.yml
.github/workflows/pypi-publish.yml
.github/workflows/python-package-check.yml
.github/workflows/pytype.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/distributed/run.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/bucket.py
bagua/torch_api/communication.py
bagua/torch_api/distributed.py
bagua/torch_api/env.py
bagua/torch_api/exceptions.py
bagua/torch_api/globals.py
bagua/torch_api/tensor.py
bagua/torch_api/utils.py
bagua/torch_api/algorithms/__init__.py
bagua/torch_api/algorithms/base.py
bagua/torch_api/algorithms/bytegrad.py
bagua/torch_api/algorithms/decentralized.py
bagua/torch_api/algorithms/gradient_allreduce.py
bagua/torch_api/algorithms/q_adam.py
bagua/torch_api/contrib/__init__.py
bagua/torch_api/contrib/fused_optimizer.py
bagua/torch_api/contrib/load_balancing_data_loader.py
docker/Dockerfile.pytorch-1.9.0-cuda10.2-cudnn7
docker/Dockerfile.pytorch-1.9.0-cuda11.1-cudnn8
docs/conf.py
docs/doc-requirements.txt
docs/index.rst
docs/_autoapi_templates/index.rst
docs/_autoapi_templates/python/attribute.rst
docs/_autoapi_templates/python/class.rst
docs/_autoapi_templates/python/data.rst
docs/_autoapi_templates/python/exception.rst
docs/_autoapi_templates/python/function.rst
docs/_autoapi_templates/python/method.rst
docs/_autoapi_templates/python/module.rst
docs/_autoapi_templates/python/package.rst
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