Programming · Noah Johansson · 2026-10-20

Google abre código fuente TorchTPU, un backend nativo de PyTorch para TPUs

Google ha abierto código fuente TorchTPU, un backend nativo de PyTorch para TPUs. Durante años, usar PyTorch en TPUs significaba pasar por PyTorch/XLA, lo cual funcionaba pero siempre se sintió como un camino de segunda clase comparado con JAX. Un backend nativo podría hacer que los TPUs sean una opción mucho más realista para el gran número de equipos que trabajan con PyTorch. ¿Alguien lo ha probado ya y esto cambia su cálculo sobre GPU vs TPU?

52 ✦ · 18 💬

Únete a la conversación

Comentarios

Amelia Hughes · 11 ✦

¿alguien sabe qué pasa con pytorch/xla ahora? ¿en desuso o coexistiendo?

Mango Baker · 9 ✦

el nombre de la organización de GitHub google-pytorch me resulta gracioso por alguna razón

Zeynep Demir · 7 ✦

si esto realmente funciona sin las habituales sorpresas de compilación de xla, es algo grande para cualquiera que esté fuera del precio de nvidia

Diego Quispe · 6 ✦

lo intenté durante una hora, el ciclo de entrenamiento básico funcionó. Las operaciones personalizadas son donde espero que las cosas se rompan

Mina Chowdhury · 4 ✦

la verdadera pregunta es la disponibilidad. TPUs solo en Google Cloud, así que esto también es un embudo de ventas en la nube