colossalai.nn.optimizer.colossalai_optimizer