tilelang

关于

用 Python 写高性能 GPU 算子。这是一个专为编写高性能 GPU/CPU/NPU 算子设计的领域专用语言，底层基于 TVM 编译器构建，让开发者不必手写底层代码也能实现 GEMM、FlashAttention 等算子的极致优化。它支持 NVIDIA CUDA、AMD ROCm、Apple Metal 和华为昇腾等多种硬件，内置自动调优和调试可视化工具，并提供 DeepSeek MLA 等大模型算子示例。

收录记录

1. GitHub Trending

   2026年10月2日 · 1 次收录

2. HelloGitHub·月刊推荐

   2026年10月28日 · 1 次收录