AI工具导航

TorchTPU

在谷歌规模上原生运行PyTorch于TPU

TorchTPU 是什么?

TorchTPU是谷歌为TPU设计的原生PyTorch后端。只需极少的代码改动即可运行现有PyTorch工作负载,通过融合即时执行模式可获得50-100%以上的速度提升,并能扩展至10万+芯片集群——无需静态图编译即可快速上手

产品详细介绍

在谷歌规模上原生运行PyTorch于TPU

TorchTPU是谷歌为TPU设计的原生PyTorch后端。只需极少的代码改动即可运行现有PyTorch工作负载,通过融合即时执行模式可获得50-100%以上的速度提升,并能扩展至10万+芯片集群——无需静态图编译即可快速上手

定价方案

需要付费
官网查看
定制方案
  • 暂未采集到具体套餐,价格以官网为准

查看官网价格

官网信息