谷歌与 Meta 开源 TorchTPU,PyTorch 模型可迁移至 TPU

谷歌宣布开源与 Meta PyTorch 团队共同开发的原生 TPU 软件栈 TorchTPU,开发者用较少代码改动即可把现有 PyTorch 模型部署到 TPU。

🔗 阅读原文

via 无言AIGC · https://aihot.wuyanshuo.cn/items/ndailfymxxvwul6ygill0gll0