Google open-sources TorchTPU, a native PyTorch back end for TPUs
Contribute to google-pytorch/torch_tpu development by creating an account on GitHub.
AI Summary
The GitHub project google-pytorch/torch_tpu provides a backend for running PyTorch models on Google's Tensor Processing Units (TPUs). It features custom ATen kernels, a compilation cache for optimized execution, and supports distributed training. The repository includes examples for various models and a user guide, with installation instructions compatible with Python 3.11 and above.


