Bagua

This website contains Bagua API documentation. See tutorials if you need step by step instructions on how to use Bagua.

API Documents

• bagua
  • bagua.torch_api
    • bagua.torch_api.algorithms
      • bagua.torch_api.algorithms.base
      • bagua.torch_api.algorithms.bytegrad
      • bagua.torch_api.algorithms.decentralized
      • bagua.torch_api.algorithms.gradient_allreduce
      • bagua.torch_api.algorithms.q_adam
    • bagua.torch_api.contrib
      • bagua.torch_api.contrib.fused_optimizer
      • bagua.torch_api.contrib.load_balancing_data_loader
    • bagua.torch_api.bucket
    • bagua.torch_api.communication
    • bagua.torch_api.distributed
    • bagua.torch_api.env
    • bagua.torch_api.tensor