[pytorch-lightning](https://pytorch-lightning.readthedocs.io) allows for [less boilerplate](https://pytorch-lightning.readthedocs.io/en/latest/introduction_guide.html#why-pytorch-lightning) and [more optimization](https://pytorch-lightning.readthedocs.io/en/latest/performance.html). Maybe it should be used to allow for easier reuse of the code.