Bagua API Documentation
latest

API Documents

  • bagua
    • bagua.torch_api
Bagua API Documentation
  • Bagua API Documentation »
  • bagua
  • Edit on GitHub

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.algorithms
      • bagua.torch_api.algorithms.async_model_average
      • 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.checkpoint
      • bagua.torch_api.checkpoint.checkpointing
    • bagua.torch_api.contrib
      • bagua.torch_api.contrib.fuse
      • bagua.torch_api.contrib.utils
      • bagua.torch_api.contrib.cache_loader
      • bagua.torch_api.contrib.cached_dataset
      • bagua.torch_api.contrib.load_balancing_data_loader
      • bagua.torch_api.contrib.sync_batchnorm
    • bagua.torch_api.data_parallel
      • bagua.torch_api.data_parallel.bagua_distributed
      • bagua.torch_api.data_parallel.distributed
      • bagua.torch_api.data_parallel.functional
    • bagua.torch_api.model_parallel
      • bagua.torch_api.model_parallel.moe
    • bagua.torch_api.bucket
    • bagua.torch_api.communication
    • bagua.torch_api.distributed
    • bagua.torch_api.env
    • bagua.torch_api.tensor
Next Previous

© Copyright 2021, Kuaishou AI Platform and DS3 Lab Revision a71c904b.

Read the Docs v: latest
Versions
latest
stable
v0.9.0
v0.8.2
v0.8.1.post1
v0.8.1
v0.8.0
v0.7.0
v0.6.3
v0.6.2
v0.6.1
v0.6.0
0.5.0
0.4.0
Downloads
html
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.