Comment by Akseleu_omirbay

And what about TPU support? Is it compatible with Google TPUs or strictly limited to gpu backends like CUDA and Metal?