PyTorch Lightningは、Lightning AIが提供する高レベル学習フレームワークです。PyTorchのコードを整理し、研究ロジックとエンジニアリングの詳細を分離することで、1台から1万台超のGPUまでコード変更なしにスケールできます。
PyTorchベースの学習フレームワークを知りたい相談
研究ロジックとエンジニアリングの分離という設計思想