Orbit · Explore

Maschinell übersetzt aus: Englisch · Original ansehen

Programming · Noah Johansson · 2026-10-20

Google stellt TorchTPU als Open-Source-Projekt vor, einen nativen PyTorch-Backend für TPUs

Google hat TorchTPU als Open-Source-Projekt vorgestellt, einen nativen PyTorch-Backend für TPUs. Jahrelang bedeutete die Nutzung von PyTorch auf TPUs, über PyTorch/XLA zu gehen, was funktionierte, sich aber immer wie ein zweiter Weg im Vergleich zu JAX anfühlte. Ein nativer Backend könnte TPUs zu einer viel realistischeren Option für die große Anzahl von Teams machen, die mit PyTorch arbeiten. Hat es schon jemand ausprobiert und ändert das Ihre Kalkulation bei der Wahl zwischen GPU und TPU?

52 ✦ · 18 💬

Mach mit bei der Diskussion

Kommentare

Amelia Hughes · 11 ✦

weiß jemand, was jetzt mit pytorch/xla passiert? Veraltet oder koexistierend?

Mango Baker · 9 ✦

der GitHub-Org-Name google-pytorch ist aus irgendeinem Grund lustig für mich

Zeynep Demir · 7 ✦

wenn das tatsächlich ohne die üblichen XLA-Kompilierungsüberraschungen funktioniert, ist das eine große Sache für jeden, der von Nvidia abgehängt wurde

Diego Quispe · 6 ✦

ich habe es eine Stunde lang versucht, der grundlegende Trainingsloop lief. Bei den benutzerdefinierten Operationen erwarte ich, dass Dinge kaputtgehen

Mina Chowdhury · 4 ✦

die eigentliche Frage ist die Verfügbarkeit. TPUs sind nur auf Google Cloud verfügbar, also ist dies auch ein Cloud-Verkaufskanal