Urinn

### Describe the feature As a researcher developing auto parallelism, it would be great if we can use Colossal-AI to test the parallel schedule we design. Currently, Colossal-AI supports 1D,2D,2.5D,...