跳到主内容
@wquguru
精选75Rohan Paul产品发布/更新多源精选 ×6

Google开源TorchTPU替代PyTorch/XLA

Google and Amazon now both offer native PyTorch backends that let code leave Nvi…

原文
发到 X

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

Google 和 Amazon 现在都提供原生的 PyTorch 后端,使代码能够离开 Nvidia GPU。

Google just open-sourced TorchTPU that replaces PyTorch/XLA, letting developers debug TPU models step by step.

Google 刚刚开源了 TorchTPU,它取代了 PyTorch/XLA,让开发者可以逐步调试 TPU 模型。

TorchTPU used up to 52% of peak TPU compute training Qwen 3 8B.

TorchTPU 在训练 Qwen 3 8B 时使用了高达 52% 的峰值 TPU 算力。

TPUs previously reached PyTorch through PyTorch/XLA, which queued operations for a compiler, so debugging differed from GPU work.

此前 TPUs 通过 PyTorch/XLA 接入 PyTorch,后者将操作排队交由编译器处理,因此调试方式与 GPU 工作不同。

TorchTPU replaces that layer, running each operation as written for debugging, then switching to torch.compile for production speed.

TorchTPU 取代了该层,按原样运行每个操作以便调试,然后在生产环境中切换到 torch.compile 以提升速度。

更进一步:量化金融体系

看懂新闻只是起点——沿量化金融路径,把它变成能交付的工程能力

进入量化体系 →