lightning-ai/pytorch-lightning 是一个用于预训练和微调 PyTorch 模型的深度学习框架。它将重复性的工程代码自动化,同时让模型逻辑由用户直接掌控,使训练任务能够从 CPU 环境扩展到多节点 GPU。按 GitHub 发布记录,最新版本为 2.6.6。
该项目的目标是为 PyTorch 的训练循环和分布式执行提供结构化方案。仓库说明指出,无需修改代码,即可对任意规模的模型进行预训练和微调,硬件规模从 1 个 GPU 到 10,000 个以上 GPU。
核心包含两个包:负责规模化训练与部署的 PyTorch Lightning,以及提供更细粒度控制的 Lightning Fabric。两个包都管理分布式策略、检查点、精度设置和迭代式训练运行。均以 Python 实现,采用 Apache-2.0 许可证。仓库创建于 2019 年 3 月,GitHub 星标数记录为 31423。
不过,由于本次记录中没有发布说明,2.6.6 的变更内容无法确认。也没有提供性能基准测试或与其他框架的对比资料。研究记录给出的动量指标为 10 分中的 2.6,偏低。因此,与其说这是一个新增功能的版本,它更适合作为考察训练循环与分布式执行结构的对象。