pytorch-lightning

Deep learning framework to train, deploy, and ship AI products Lightning fast.