v0.2.6
[hotfix]: Remove math.prod dependency (#2837)
fixed using zero with tp cannot access weight correctly
[NFC] polish colossalai/nn/loss/loss_2p5d.py code style (#1553)
[NFC] polish colossalai/nn/lr_scheduler/linear.py code style (#1716)
[NFC] polish colossalai/nn/metric/_utils.py code style (#1727)
[format] applied code formatting on changed files in pull request 2997 (#3008)
Rename class method of ZeroDDP (#2692)
[kernel] added jit warmup (#1792)
[NFC] polish colossalai/nn/init.py code style (#1292)