Helion on TPU: Towards Hardware Heterogeneous Kernel Authoring

TL;DR Helion is PyTorch s high-level DSL for writing performance-portable ML kernels. Partnering with Google, we have built a TPU backend that compiles Helion kernels to Pallas, providing a PyTorch-friendly way...