GitHub repository: pytorch-labs/helion
Primary language: Python
Description: A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.