TorchTPU
TorchTPU adds a native TPU backend to PyTorch. It supports Pallas kernels, distributed training, and torch.compile integration.
What it does
Developers can run and train Hugging Face models on TPUs using PyTorch, and use TPU kernels to accelerate workloads.
- Availability
- Install on a TPU VM with pip install torch-tpu
- Pricing model
- Not established
- Deployment
- TPU VM; TPUs are available on Google Cloud
- Privacy
- Not established
- Licence
- Not established
Capability
Modality
Access
Deployment
Pricing
Target user
Evidence
6 claims- TorchTPU adds TPU as a new backend to PyTorch. [1]
TorchTPU adds TPU as a new backend to PyTorch.
Hugging Face. Quoted text is outside CC BY 4.0 and keeps its original rights. - It includes Pallas kernels tuned for TPUs, distributed training support, and torch.compile integration. [1]
It comes with Pallas kernels tuned for TPUs, support for distributed training, and a torch.compile integration for when you want more speed.
Hugging Face. Quoted text is outside CC BY 4.0 and keeps its original rights. - The article provides a pip command to install TorchTPU on a TPU VM. [1]
Install TorchTPU, grab a model from the Hub, and send it to "tpu" .
Hugging Face. Quoted text is outside CC BY 4.0 and keeps its original rights. - Transformers models can be loaded on TPU with device_map="tpu". [1]
Now that TorchTPU brings native support, Optimum TPU is not required anymore to run Hugging Face models with Transformers and Diffusers.
Hugging Face. Quoted text is outside CC BY 4.0 and keeps its original rights. - Diffusers pipelines can run on TPU using pipe.to("tpu"). [1]
- The article reports that 96.5% of the Transformers tests that run on TPU pass. [1]
Of the tests that run on TPU, 96.5% pass.
Hugging Face. Quoted text is outside CC BY 4.0 and keeps its original rights.
Sources
1
1
Verified
Versions
7| Version | Effective | Change | Changed fields | State |
|---|---|---|---|---|
| 7 | 2026-10-28 00:02 UTC | material change | capability, modality, access, deployment | Current |
| 6 | 2026-10-26 01:12 UTC | material change | capability, modality, access, deployment | Current |
| 5 | 2026-10-25 01:09 UTC | material change | access, deployment | Current |
| 4 | 2026-10-24 01:29 UTC | material change | capability, modality, access | Current |
| 3 | 2026-10-23 01:10 UTC | material change | capability, modality, deployment | Current |
| 2 | 2026-10-22 00:28 UTC | material change | capability, modality, access, deployment | Current |
| 1 | 2026-10-20 17:18 UTC | publication | None | Current |
In briefs
2- A toolkit for voice agents that listen, respond, and take action Weekly brief 2026-W43
- Voice agents can take actions during a live conversation Daily brief 2026-10-20
Original catalog content is licensed under CC BY 4.0, attributed to wowsuchai. Source quotations and linked content keep their original rights.