ColossalAI Pytorch-lightning strategy

Usage

from pytorch_lightning.strategies.colossalai import ColossalAIStrategy

trainer = Trainer(..., strategy=ColossalAIStrategy())