google-pytorch / torch_tpu
git clone https://github.com/google-pytorch/torch_tpu.git
🔥 on GitHub Trending today
Provides a PyTorch backend that runs PyTorch models and operations on Google's Tensor Processing Units (TPUs).
Why you might care
Targets developers who want to run existing PyTorch code on TPU hardware, with custom ATen kernels, a compilation cache, torch.compile integration, and distributed training support (data and tensor parallelism). Includes example models such as Llama, ResNet, and DLRM.
+67
stars, 24h · #590
+~469
stars, 7 days · #468
+~2.0k
stars, 30 days · #405
659
★ total · +247% this week
45
forks · 268 open issues
Star history (as recorded here)
LanguageC++
License—
Created2026-03-09 (7mo ago)
Last pushtoday
Watchers3
OwnerOrganization
Seen on GitHub Trending
- 2026-10-23 — #4 daily in c++ · +104 ★
Links
- GitHub: github.com/google-pytorch/torch_tpu
- Homepage: https://torchtpu.readthedocs.io
- Releases · Issues · Hacker News mentions
README
Loading README…
Blurb generated by claude-haiku-5-5 today. Star deltas marked ~ are extrapolated from partial history.