colossalai.nn.optimizer ======================= .. automodule:: colossalai.nn.optimizer :members: .. toctree:: :maxdepth: 2 colossalai.nn.optimizer.colossalai_optimizer colossalai.nn.optimizer.cpu_adam colossalai.nn.optimizer.fused_adam colossalai.nn.optimizer.fused_lamb colossalai.nn.optimizer.fused_sgd colossalai.nn.optimizer.hybrid_adam colossalai.nn.optimizer.lamb colossalai.nn.optimizer.lars