AI圈报
产品发布 / 更新精选

Google 与 Meta 发布 TorchTPU:PyTorch 原生 TPU 后端

信息来源:PyTorch:Blog(RSS)·
原始标题:Introducing TorchTPU

内容摘要

Google 与 Meta 联合开源 TorchTPU,为 Google TPU 提供原生 PyTorch 后端,模型只需把 .to("cuda") 改成 .to("tpu") 即可开箱运行,无需 JAX 依赖。
内容层级精选情报
发布时间(北京时间)
本站收录时间(北京时间)
信息来源PyTorch:Blog(RSS)
站内情报编号intel-22da5698ef2e093849650b2b