TorchTPU

谷歌与Meta PyTorch团队联合开发并开源的原生TPU软件栈，PyTorch模型只需少量改动（device_map="tpu"）即可跑在TPU上。

AI编程免费 ★ 4.5/5

访问官网 →

相关评测文章

谷歌开源TorchTPU解读：PyTorch代码改一行就上TPU

谷歌与Meta联合开源TorchTPU，PyTorch模型少量改动即可跑在TPU上，8000颗Ironwood验证、MFU最高52%。解读技术细节与生态影响。...

工具信息

官网	github.com
分类	AI编程
费用	有免费版
评分	★ 4.5 / 5.0
浏览	18 次

同类工具

JetBrains Mellum2