资讯

Google 开源 TorchTPU：PyTorch 原生跑上 TPU

Google 与 Meta 合作开源 TorchTPU，为 TPU 提供原生 PyTorch 后端，少量代码修改即可迁移，支持即时执行、编译执行与分布式计算。