A TorchTPU natív PyTorch-hátteret hoz a Google TPU-ihoz

A TorchTPU csapatának bejelentése 2026. október 20-án mutatja be a projektet, és jelzi, hogy a fejlesztés folytatódik. A telepített csomaggal rendelkező TPU-gépen a PyTorch importálása automatikusan betölti a végrehajtási hátteret. A csapat a cuda eszközcél tpu értékre cserélésével szemlélteti az átállást. Arra is figyelmeztet, hogy ez önmagában nem ad csúcsteljesítményt: a modell adatalakjai, műveletei és eszközök közötti felosztása továbbra is GPU-ra szabott feltételezéseket követhetnek.

A projekt kódtára Python virtuális környezetben dokumentálja a telepítést, amelyhez a Python 3.11-es vagy későbbi változatát támogatja. A kiadás telepítési parancsa a torch_tpu 0.2.0 verzióját választja ki, és megadja a PyTorch CPU-csomagjainak indexét. Ennek elhagyásakor a PyTorch CUDA-változata települ, hacsak a CPU-változatot nem telepítik külön. Az ismerkedéshez a kódtár több eszközön végzett tanításhoz és modellek futtatásához is tartalmaz példákat. Interaktív Marimo-jegyzetei TPU virtuális gépről futnak, és böngészőben nyílnak meg; a felhasználói útmutató ezek statikus változatát adja.

A nyilvános API-k dokumentációja megkülönbözteti a hosszú távon támogatott programozási felületeket a kísérletiektől. A stabil felületek visszafelé nem kompatibilis módosításai kivezetési folyamaton mennek keresztül. A kísérleti felületek e folyamat nélkül változhatnak vagy eltűnhetnek, használatukkor egyszeri figyelmeztetés jelenik meg. A torch.accelerator.is_available például stabilként szerepel, a torch.tpu.Event viszont kísérleti. Az oldal figyelmeztet, hogy más TorchTPU API-k használata nem támogatott; a TPU-kon megvalósított PyTorch-felületeket a PyTorch saját dokumentációja ismerteti.