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

Google has open-sourced TorchTPU, a native PyTorch backend that enables users to run PyTorch models and operations on Google's Tensor Processing Units (TPUs). The project includes custom ATen kernels, a compilation cache for optimized execution, support for distributed training, and integration with torch.compile() for graph mode execution. Users can install TorchTPU via pip and access comprehensive documentation and interactive notebooks for getting started with TPU-accelerated PyTorch development.