v0.3.1
[test] fixed the triton version for testing (#2608)
[bf16] add bf16 support (#3882)
[NFC] fix typo colossalai/amp auto_parallel autochunk (#3756)
[NFC] polish colossalai/amp/__init__.py code style (#3272)
Develop/experiments (#59)