Google libera TorchTPU, el backend nativo de PyTorch para TPUs

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

Google ha publicado TorchTPU, un backend de código abierto que permite ejecutar modelos y operaciones de PyTorch directamente en sus Tensor Processing Units. Incluye kernels ATen personalizados, caché de compilación para optimizar la ejecución, soporte para entrenamiento distribuido con paralelismo de datos y de tensores, e integración con torch.compile(). Ya está disponible en PyPI (versión 0.2.0) y requiere Python 3.11 o superior.