Google padara TorchTPU par atvērtu kodu un vilina PyTorch izstrādātājus uz saviem čipiem

Google otrdien, 20. oktobrī, konferencē PyTorch Conference 2026 paziņoja, ka TorchTPU ir publicēts GitHub kā atvērtā koda projekts. Tā ir PyTorch aizmugursistēma (backend), kas ļauj ar šo populāro MI izstrādes rīkkopu veidotus modeļus apmācīt un darbināt uz Google pašu izstrādātajiem MI paātrinātājiem, tā sauktajiem TPU (Tensor Processing Unit).

Līdz šim lielākā daļa MI lietotņu pēc noklusējuma tiek rakstītas Nvidia grafiskajiem procesoriem. PyTorch sākotnēji izstrādāja Meta, un daudzas tā funkcijas tika veidotas tieši Nvidia aparatūrai, norāda TechCrunch. Izdevums Google soli vērtē kā mēģinājumu kopā ar Amazon mazināt Nvidia priekšrocības izstrādātāju vidē. Amazon tikai dienu iepriekš laida klajā līdzīgu rīku TorchNeuron saviem Trainium čipiem.

Pēc Google un PyTorch projekta skaidrojuma TorchTPU izstrāde sākās 2025. gada vidū sadarbībā ar Meta PyTorch komandu. Esošu PyTorch kodu var pārcelt uz TPU ar nelielām izmaiņām, piemēram, ierīces norādi "cuda" nomainot uz "tpu". Rīks atbalsta gan interaktīvo (eager) režīmu kļūdu meklēšanai, gan kompilēšanu ar torch.compile un sadalītu apmācību vairākos čipos. Iepriekšējais risinājums PyTorch/XLA prasīja koda pārveidošanu un ilgi kompilēja, atzīst paši izstrādātāji.

Google sadarbojās arī ar citiem atvērtā koda projektiem. Hugging Face bibliotēkās Transformers un Diffusers modeļus tagad var palaist uz TPU, norādot ierīci "tpu", un uzņēmums savā blogā raksta, ka atsevišķs rīks Optimum TPU šim nolūkam vairs nav vajadzīgs. TPU atbalsts pievienots arī modeļu apkalpošanas dzinējiem vLLM un SGLang, apmācības rīkam TorchTitan un uzdevumu orķestrēšanas sistēmai Ray.

Papildus Google paziņoja, ka ierīcēs darbināmo modeļu rīks LiteRT (agrāk TensorFlow Lite) pievienojas PyTorch ekosistēmai, bet TPU aģentu komplekts, kas palīdz pārcelt modeļus uz Google čipiem un labot kļūdas, ir pieejams kā agrīns priekšskatījums. Uzņēmums apgalvo, ka TorchTPU jau tiek lietots ražošanā dažiem lielākajiem Google Cloud klientiem.

TechCrunch norāda, ka jaunie rīki paši par sevi nenozīmē, ka izstrādātāji masveidā atteiksies no Nvidia aparatūras. Amazon gan iet vēl tālāk un iesniedz izmaiņas arī pamata PyTorch projektā, lai daļu funkciju atsaistītu no Nvidia čipiem. Tas, vai šīs izmaiņas tiks pieņemtas, noteiks, cik viegli nākotnē PyTorch kodu varēs pārnest starp dažādu ražotāju aparatūru un mākoņpakalpojumiem.