pytorch-lightning

f3252d92 - Merge branch 'feature/rewrap-optimized-module' of github.com:Lightning-AI/lightning into feature/rewrap-optimized-module