tilelang
Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels
Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels
Train, fine-tune, and evaluate models — the infrastructure layer.
Key features
- Python 风格写 kernel
- 底层 TVM 编译优化
- 支持 CUDA/Metal/昇腾
- GEMM/Attention 开箱即用
- 配套 LSP 与解谜教程
- 源码级编译诊断
Good for
- 手写高性能算子
- FlashAttention 优化
- 国产 NPU 适配
- 算子开发学习
Stack & tags
PythonTVMCUDAMetalPyPI
Quick start
pip install tilelangFrom the project README on GitHub.
Similar open-source projects
tensorflow
★ 200,721An Open Source Machine Learning Framework for Everyone
transformers
★ 167,239🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.