I want to implement Colossal Ai distributed training in my training script, what are the steps needed to convert my pytorch training script into using the Colossal Ai distributed training library?