Lightning AI

Deep learning framework to train, deploy, and ship AI products Lightning fast.

通过Lightning框架，我们可以更专注于“调整model内部结构”和“数据工程”，提升算法开发与训练的效率。

很多大厂内部的训练框架，无非做两件事：

• 方便调用公司内部集群进行多机多卡的训练
• 方便调用公司内部各类数据存储单元，喂数据给到模型

而Lightning框架开源可学习，远比国内一些大厂的训练框架更简洁高效，而且说明文档/教程更加全面、易用。

借此，我们能做到

• 轻松实现单机多卡、多机多卡的训练
• 轻松接入deepspeed训练大模型
• 不用关注model.to(device)等等细节，避开device相关冲突，腾挪数据的操作全部由框架保证
• 通过Trainer控制各类参数，如earlystop、epoch、pruning、tensorboard&log等

Lightning的强大，不仅仅是以上内容就能概括的。而且它与Pytorch特性完美融合，使用过程丝滑毫无尿点。