Comment by kjkjhgkjyj

TensorFlow and PyTorch support TPUs. It's pretty painless.