机器之心 (微信公众号)T2·88 分
谷歌开源 TorchTPU,TPU 成 PyTorch 原生设备
原标题:谷歌开源TorchTPU!TPU成PyTorch原生设备,CUDA最深的护城河开始松动
- TorchTPU 让 TPU 与 GPU、CPU 并列,成为 PyTorch 的原生支持设备。
- 开发者可通过 Hugging Face Transformers v5 中的 device_map="tpu" 参数一键部署模型。
- 此举旨在解决 PyTorch/XLA 转接层带来的高工程成本,挑战 CUDA 生态主导地位。