ColossalAI

ColossalAI: Open-Source Large-Scale AI Training Framework

Training large AI models is fundamentally a distributed computing problem. A single 70B parameter model requires more memory than any GPU can...