Train deep learning models with PyTorch Lightning.

Train models with PyTorch Lightning in ZenML

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.