跳到正文
Rohan Paul (X)· rohanpaul_ai·· 13 小时前AI 评分68

Google 开源 TorchTPU,替换 PyTorch/XLA 支持逐步调试 TPU 模型

Google and Amazon now both offer native PyTorch backends that let code leave Nvidia GPUs.

AI 导读

Google 开源 TorchTPU,用来替换 PyTorch/XLA,让开发者可以逐步调试 TPU 模型。作者称 TorchTPU 在训练 Qwen 3 8B 时用到了最高 52% 的 TPU 峰值算力。

来源:Rohan Paul (X) · x.com