Colossal-AI一个整合高效并行技术的 AI 大模型训练系统,具有高效并行化技术的集成大规模模型训练系统。最重要的是,它是完全开源的,只需进行少量修改,即可在单个消费级显卡上使用更大的模型训练现有的深度学习项目,让每个人都可以在家中训练大型 AI 模型!特别是,它使大型 AI 模型微调和推理等下游任务和应用程序部署变得更加容易!

单 GPU 演示 GPT-2:相同硬件上的模型尺寸大 20 倍


单 GPU 演示 GPT-2:相同硬件上的模型尺寸大 20 倍


开源并行技术的 AI 大模型训练系统地址

官网网站:

https://www.colossalai.org/zh-Hans/

GitHub地址:

https://github.com/hpcaitech/ColossalAI