0
votes

I'm trying to run PyTorch on TPU's. However I am unable to determine how to get a compilation with the XLA Device.
The PyTorch precompiled versions don't have XLA and the from source instructions don't ether. Is there any way to do it without going threw torch_xla.