bagua

Bagua is a communication library developed by Kuaishou Technology and DS3 Lab for deep learning.

See tutorials for Bagua’s rationale and benchmark.

Subpackages

• bagua.torch_api
  • bagua.torch_api.contrib
    • bagua.torch_api.contrib.fused_optimizer
    • bagua.torch_api.contrib.load_balancing_data_loader
  • bagua.torch_api.communication
  • bagua.torch_api.distributed
  • bagua.torch_api.env

Submodules

• bagua.bagua_define