Running PyTorch on TPU: a bag of tricks