Google open-sources TorchTPU, a native PyTorch back end for TPUs
Google has open-sourced TorchTPU, a native PyTorch back end designed for TPUs, allowing developers to run PyTorch models natively on TPU hardware.
Key points
- •It is developed as a native back end to execute PyTorch models on TPUs.
- •Development contributions are facilitated through the google-pytorch/torch_tpu repository on GitHub.