pytorch-lightning
PyTorch Lightning
helps organize PyTorch code and decouple the science code from the
engineering code. It’s more of a style-guide than a framework. By
organizing PyTorch code under a LightningModule
, Lightning makes
things like TPU, multi-GPU and 16-bit precision training (40+ other
features) trivial.
For more information, please see: