Google just open-sourced TorchTPU, a native PyTorch backend for TPUs (52% MFU on Qwen 3 8B)
This story is from 2026-10-20. It is preserved in the archive; the latest stories are on the live feed.
Google open-sourced TorchTPU today at PyTorch Conference 2026. It is a PyTorch backend for Cloud TPUs, co-developed with Meta's PyTorch team, and it replaces PyTorch/XLA. You target a TPU with torch.device("tpu") , run eager by default, and use torch.compile(backend="tpu") for production speed. Wha…
Read the full story at r/machinelearningnews ↗
Timeline · 1 report
- 2026-10-20 15:27 · r/machinelearningnews
Google just open-sourced TorchTPU, a native PyTorch backend for TPUs (52% MFU on Qwen 3 8B)