PyTorch-Lightning and fastai Checking out PL and fastai here Tried out TPU for lightning works pretty well for larger batch sizes (trained on MNIST), not so well for smaller ones