推論系統
PyTorch Helion Adds TPU Backend, Bridging Triton and Pallas with the Same High-Level Kernels
PyTorch and Google have built a TPU backend for Helion, allowing PyTorch-style kernels to compile to Pallas and reducing the need to write separate low-level TPU kernels. The team measured up to 838 TFLOPs for Flash Attention on TPU v7, though the feature remains under active development.