ColossalAI Pytorch-lightning strategy Usage from pytorch_lightning.strategies.colossalai import ColossalAIStrategy trainer = Trainer(..., strategy=ColossalAIStrategy())