colossalai.zero.init\_ctx ========================= .. automodule:: colossalai.zero.init_ctx :members: .. toctree:: :maxdepth: 2 colossalai.zero.init_ctx.init_context