Código abierto
Google publica en GitHub TorchTPU, un backend nativo de PyTorch para TPU
El proyecto permite ejecutar modelos de PyTorch en las unidades de procesamiento tensorial de Google. Incluye kernels personalizados, caché de compilación y ejemplos de entrenamiento distribuido e inferencia.
Por Redacción Pico Tecno|