colossalai.zero.sharded\_optim.bookkeeping
==========================================

.. automodule:: colossalai.zero.sharded_optim.bookkeeping
   :members:


.. toctree::
   :maxdepth: 2

   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