Public KG record
TorchTPU
Recorded overview
Open-sourced library that runs PyTorch and Hugging Face models natively on TPUs.
An evidence-backed entity from the project knowledge graph; detailed tool metadata is still being added.
Recorded details
- Benchmarks: 持续测试约 5,000 个 Hugging Face 模型
- Capabilities: 支持 MPMD 分歧执行
- Capabilities: 支持 DDP、FSDP2、DTensor
- Capabilities: 支持分布式训练
- Capabilities: 提供针对 TPU 调优的 Pallas kernels
- Deployment: 运行于 Google Cloud TPU
- Deployment: Runs on Google Cloud TPUs
- Integrations: Transformers v5 可用 device_map="tpu" 部署
Evidence coverage
The KG currently records 9 known information dimensions and has 9 undocumented dimensions, linked to 3 public source articles. These counts describe record coverage, not model performance or verified fit.
Related public records
- Google Cloud — develops
- Meta — develops
- TorchTitan — uses
- Google Cloud TPU — applied to
- Ironwood (TPU7x) — applied to