colossalai

An integrated large-scale model training system with efficient parallelization techniques


License
Apache-2.0
Install
pip install colossalai==0.1.11rc4