Running PyTorch on TPU: a bag of tricks

Open in new window