AI product Open source

TileLang

TileLang is a Pythonic domain-specific language and compiler for developing high-performance GPU, CPU, and NPU kernels such as GEMM, quantized matrix multiplication, FlashAttention, and linear attention. Built on TVM, it uses tiled programming primitives for memory transfers, pipelining, matrix operations, parallel elementwise work, layouts, and target-specific code generation; its JIT specializes kernels for input shapes and compile-time arguments on first use. The project supports CUDA, ROCm/HIP, Apple Metal, Huawei Ascend, and experimental LLVM, CuTe DSL, and WebGPU backends, with additional ecosystem adapters for other accelerator platforms. It is distributed as a Python package through PyPI, with source builds and nightly wheels available.

View repository Visit site