tile-ai / tilelang

A Python-embedded domain-specific language and TVM-based compiler for writing high-performance GPU, CPU, and NPU kernels.

pytorch / pytorch

Python library providing GPU-accelerated tensor computation like NumPy, plus a tape-based autograd system for building deep neural networks.