- Category
- Developer Tools
- Rank
- No. 1105Tools index
- Pricing
- Open Source
- Type
- TOOL
- GitHub
- 404 stars
- Date
About
A PyTorch backend that lets users run PyTorch models and operations on Google TPU devices. It includes custom ATen kernels, a compilation cache for XLA computations, data and tensor parallel training, torch.compile integration, and examples for models such as Llama, Qwen, ResNet and UNet.
Intel on TorchTPU
Tags
pytorchtpuxlamldistributed-traininggoogle
Tech Stack
Python
Comments (0)
No comments yet
Editorially curated, with community endorsements as a secondary signal. Corrections welcome.