Google lanserer TorchTPU for å kjøre PyTorch-kode på TPU-brikker
25. oktober 2026 · 1 min lesing
Google har lansert TorchTPU, som er en bakende for PyTorch-biblioteket som gjør det mulig å kjøre kode direkte på TPU-brikker med full ytelse.
Utviklingen av TorchTPU er et samarbeid mellom Google og PyTorch-teamet fra Meta. Målet med prosjektet er å utvide bruken av PyTorch til nye brikker utover tradisjonelle CPU-er og GPU-er. Dette ligner på hvordan Amazon har utviklet TorchNeuron for å kompilere PyTorch-kode til Trainium-brikker.
Google har samarbeidet med Hugging Face for å teste ytelsen til flere åpne modeller på TorchTPU. Det er også gjennomført tester på TPU med kjente biblioteker som vLLM, SGLang, llm-d, TorchTitan, VeRL, Ray og LiteRT. Utviklingen av TorchTPU startet i 2025, og verktøyet ble denne uken lansert som fullstendig åpen kildekode på GitHub under PyTorch Conference 2026.