Dev Tools · via Hacker News

Google open-sources TorchTPU, a native PyTorch back end for TPUs

PyTorch code can target Google TPUs directly, making TPUs a more realistic option outside JAX.