v0.3.1
[doc] Fix typo under colossalai and doc(#3618)
[CheckpointIO] a uniform checkpoint I/O module (#1689)
fix typo with colossalai/trainer utils zero (#3908)
[setup] support pre-build and jit-build of cuda kernels (#2374)
[NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779)
[test] refactor tests with spawn (#3452)
[utils] Add use_reetrant=False in utils.activation_checkpoint (#1460)
[utils] refactor parallel layers checkpoint and bcast model on loading checkpoint (#1548)
[refactor] refactor the memory utils (#715)
[gemini] APIs to set cpu memory capacity (#809)
Refactored docstring to google style