Train deep learning models with PyTorch.

Train models with PyTorch in ZenML

PyTorch is one of the most popular libraries for the training and inference of deep neural networks. PyTorch is especially popular among the machine learning and deep learning research communities, and new developments from research are often accompanied by open-source PyTorch code. With ZenML's PyTorch integration, you can load, train, and deploy PyTorch models within your ZenML pipelines.