pip install colossalai-nightly