PyTorch Lightning — PyTorch 고수준 학습 프레임워크

PyTorch Lightning — PyTorch 고수준 학습 프레임워크

PyTorch Lightning은 "배터리 포함(batteries included)" 딥러닝 프레임워크로, 최대한의 유연성을 원하면서도 스케일에서 성능을 극대화하고 싶은 AI 연구자·머신러닝 엔지니어를 위한 도구예요. PyTorch 코드를 조직화해서 보일러플레이트를 없애고 확장성을 여는 것이 핵심입니다.

출처: https://lightning.ai/docs/pytorch/stable/starter/introduction.html

Lightning은 raw PyTorch로 아이디어를 시도하되 보일러플레이트 없이 최대 유연성을 주고, 연구·엔지니어링 코드를 분리해 재현성과 가독성을 높이며, 코드 변경 없이 멀티 GPU/TPU/HPU 훈련과 내장 테스트를 지원해요. 아래 페이지에서 LightningModule, Trainer, DataModule을 살펴볼게요.