Google open-sources TorchTPU for PyTorch on TPUs
Google released TorchTPU as an open-source PyTorch backend for its Tensor Processing Units, making it possible to run PyTorch models and operations on TPU devices. The project packages custom ATen kernels, a compilation cache, and examples for distributed…
