Google与Meta开源TorchTPU,为Google TPU带来原生PyTorch支持

Google与Meta开源TorchTPU,为谷歌TPU提供原生PyTorch后端,支持即时执行、编译执行和分布式计算。Google称其可减少代码改动,并在TPUv7上展示了大模型推理与训练性能;项目仍在完善动态形状、编译性能等功能。

来源:InfoQ