Helion

Helion 是一种用于编写机器学习算子的 Python 嵌入式领域特定语言 (DSL),旨在编译为 Triton(一种用于 GPU 及其他设备编程的高性能后端)。相较于 Triton,Helion 旨在提高抽象级别,在实现自动调优过程更多自动化的同时,使编写正确且高效的算子变得更加容易。

Helion 是一种专为编写高性能机器学习算子而设计的 PyTorch 原生 DSL。它极大地简化了编写快速、可扩展 ML 算子的难度,并解决了高层级生产力与底层控制之间的二元对立问题。Helion 建立了一个全新的抽象层,架起了 PyTorch 用户友好性与底层语言高性能之间的桥梁。其核心创新在于:

  • PyTorch 原生语法:使用熟悉的 PyTorch 惯用法编写算子
  • 先进的自动调优:自动探索广阔的算子配置空间
  • 硬件可移植性:单一算子定义即可在 NVIDIA、AMD、Intel GPU 及其他硬件加速器上运行

Helion 由 Meta 于 2026 年 3 月捐赠给 Linux 基金会。