Colossal-AI
latest

API REFERENCE

  • colossalai
    • colossalai.amp
    • colossalai.builder
    • colossalai.communication
    • colossalai.context
    • colossalai.engine
    • colossalai.kernel
    • colossalai.logging
    • colossalai.nn
    • colossalai.registry
    • colossalai.testing
    • colossalai.trainer
    • colossalai.utils
    • colossalai.zero
    • colossalai.global_variables
    • colossalai.initialize
Colossal-AI
  • »
  • colossalai »
  • colossalai.zero »
  • colossalai.zero.sharded_optim »
  • colossalai.zero.sharded_optim.bookkeeping
  • Edit on GitHub

colossalai.zero.sharded_optim.bookkeeping

  • colossalai.zero.sharded_optim.bookkeeping.base_store
  • colossalai.zero.sharded_optim.bookkeeping.bucket_store
  • colossalai.zero.sharded_optim.bookkeeping.gradient_store
  • colossalai.zero.sharded_optim.bookkeeping.parameter_store
  • colossalai.zero.sharded_optim.bookkeeping.tensor_bucket
Previous Next

© Copyright 2022, HPC-AI Tech. Revision 6a3f9fda.

Built with Sphinx using a theme provided by Read the Docs.