Comment by AnggaSP

TPUs are not that portable and easy for both inferencing and training. It has since improved a lot with their effort on the torch backend (XLA/TorchTPU) and JAX though.

But as far as i know it currently supports just that + tensorflow (which nobody uses it anymore, least here). And last we tried, so much of our kernels needs rework that it’s not worth the effort.

This may change since ironwood but we haven’t tried that generation.