TorchTPU

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

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