PyTorchLightning/pytorch-lightning

The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate. - PyTorchLightning/pytorch-lightning