v0.3.1
[doc] Fix typo under colossalai and doc(#3618)
[shardformer] integrated linear 1D with dtensor (#3996)
[nfc] fix typo colossalai/nn (#3887)
[NFC] polish colossalai/nn/lr_scheduler/linear.py code style (#1716)
[NFC] polish colossalai/nn/metric/_utils.py code style (#1727)
[nfc]fix typo colossalai/pipeline tensor nn (#3899)
[kernel] added jit warmup (#1792)
[NFC] polish colossalai/nn/init.py code style (#1292)