pytorch-lightning

PyTorch Lightning 是一个用于简化深度学习模型训练与微调的开源框架，基于 PyTorch 构建。它通过抽象训练循环、设备适配（CPU/GPU/TPU）、混合精度、分布式训练（单机或多节点多GPU）等底层工程细节，使用户只需专注模型逻辑，无需修改代码即可在1台至数千台GPU上扩展训练。支持预训练、微调任意规模AI模型，并提供标准化模块接口（LightningModule）和可插拔策略。适用于科研实验、工业级模型训练及大规模AI研发流程。