Last I've checked (a year or two ago) PyTorch support for TPU's were atrocious. Have they gotten any better?
Last I've checked (a year or two ago) PyTorch support for TPU's were atrocious. Have they gotten any better?