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으로 낮다. 그래서 새 기능이 추가된 버전이라기보다, 학습 루프와 분산 실행 구조를 살펴볼 대상에 가깝다.