Torchtpu - GoKawiil

Google releases TorchTPU, an open-source PyTorch version for its own AI chips

Google announced TorchTPU, a version of the PyTorch framework built to run on its Tensor Processing Units, letting developers shift existing PyTorch-based AI apps to Google Cloud with a simple configuration change rather than a rewrite. The move follows Amazon's release a day earlier of TorchNeuron, its own PyTorch variant for AWS's Trainium chips, and Amazon's submission of code to decouple PyTorch features from Nvidia hardware.

techcrunch.com · 2026-10-20

Google open-sources TorchTPU, a PyTorch backend for its TPU chips

Google has released TorchTPU, an open-source project that lets PyTorch models run directly on its Tensor Processing Units. The package includes custom ATen kernels, a compilation cache for optimizing XLA computations, support for distributed training, and sample implementations for models like Llama, Qwen, ResNet and UNet. It is available via pip as version 0.2.0 or can be built from source using Bazel.

github.com · 2026-10-20