InfoQ · AI/ML/Data· Sergio De Simone·· 6 天前AI 评分62
Google 与 Meta 开源 TorchTPU，为 Google TPU 带来原生 PyTorch 支持
Google and Meta Open-Source TorchTPU to Bring Native PyTorch Support to Google TPUs
AI 导读
Google 与 Meta 开源 TorchTPU 项目，为 Google TPU 提供原生 PyTorch 后端，PyTorch 代码只需将 cuda 设备换成 tpu、nccl 后端换成 tpu_dist 即可迁移。
来源：InfoQ · AI/ML/Data · infoq.com