Google lanza TorchTPU de código abierto para facilitar la migración de apps PyTorch

Google anunció que TorchTPU, su versión de PyTorch, ya está disponible como proyecto de código abierto. La herramienta permite ejecutar aplicaciones de IA basadas en PyTorch en las unidades de procesamiento tensorial (TPU) de Google y trasladarlas a Google Cloud con un simple cambio de configuración, sin reescribir el código. PyTorch es un conjunto de herramientas de código abierto usado para crear, entrenar y ejecutar modelos de IA, y muchas de sus funciones fueron desarrolladas para trabajar con GPU de Nvidia.

Amazon anunció el lunes TorchNeuron, que ofrece una migración similar a su chip Trainium y AWS, también sin reescribir las aplicaciones. Amazon además planea aportar código al proyecto principal de PyTorch para reducir la dependencia de sus funciones del hardware de Nvidia. Google dice que trabajó con Meta, Hugging Face, Inferact y RadixArk para hacer que TorchTPU sea compatible con herramientas populares, y lanzó una suite de agentes para ayudar a migrar aplicaciones y corregir errores.