公开知识图谱记录
TorchTPU
记录摘要
通过 PrivateUse1 接入 PyTorch，支持 eager 与 torch.compile。
来自项目知识图谱的有来源实体；具体工具属性仍需补齐。
已记录信息
- 基准测试: 持续测试约 5,000 个 Hugging Face 模型
- 能力: 支持 MPMD 分歧执行
- 能力: 支持 DDP、FSDP2、DTensor
- 能力: 支持分布式训练
- 能力: 提供针对 TPU 调优的 Pallas kernels
- 部署方式: 运行于 Google Cloud TPU
- 部署方式: Runs on Google Cloud TPUs
- 集成: Transformers v5 可用 device_map="tpu" 部署
资料覆盖
知识图谱当前记录了 8 个已知资料维度，另有 9 个维度尚未记录，关联 2 篇公开资料。这些数字只描述资料覆盖，不代表模型性能或适配性已经验证。
相关公开记录
- Google Cloud — 研发
- Meta — 研发
- TorchTitan — 采用
- Google Cloud TPU — 用于
- Ironwood (TPU7x) — 用于