PyTorch Lightning

PyTorch Lightning is a framework that helps you organize and scale your PyTorch code and it has recently become very popular among machine learning and deep learning researchers. With ZenML's PyTorch integration, you can load, train, and deploy PyTorch Lightning models within your ZenML pipelines.