pytorch-lightning

5d7f8564 - Merge branch 'master' into feature/rewrap-optimized-module