Google 和 Meta 开源了 TorchTPU,这是一个面向 Cloud TPU 的原生 PyTorch 后端。它默认以 eager 模式运行,通过 torch.compile 和 XLA 进行编译,并取代 PyTorch/XLA 用于新工作负载。文章 Google Open-Sources TorchTPU, a Native PyTorch Backend for Cloud TPUs 首次发布于 MarkTechPost 。
news
谷歌开源TorchTPU,这是云TPU的原生PyTorch后端
Google 和 Meta 开源了 TorchTPU,这是一个面向 Cloud TPU 的原生 PyTorch 后端。它默认以 eager 模式运行,通过 torch.compile 和 XLA 进行编译,并取代 PyTorch/XLA 用于新工作负载。文章 Google Open-Sources TorchTPU, a Native PyTorch Backend for Cloud TPUs 首次发布于 MarkTechPost 。