sustcsonglin / fla-tilelang
☆19Updated 3 weeks ago
Alternatives and similar repositories for fla-tilelang:
Users that are interested in fla-tilelang are comparing it to the libraries listed below
- XAttention: Block Sparse Attention with Antidiagonal Scoring☆118Updated this week
- Quantized Attention on GPU☆45Updated 4 months ago
- ☆72Updated last week
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆66Updated last week
- ☆68Updated 2 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆68Updated 9 months ago
- 🔥 A minimal training framework for scaling FLA models☆92Updated last week
- ☆19Updated 3 months ago
- 16-fold memory access reduction with nearly no loss☆86Updated last week
- ☆65Updated last month
- PyTorch bindings for CUTLASS grouped GEMM.☆77Updated 5 months ago
- SQUEEZED ATTENTION: Accelerating Long Prompt LLM Inference☆45Updated 4 months ago
- Transformers components but in Triton☆32Updated 2 weeks ago
- ☆122Updated last month
- Efficient triton implementation of Native Sparse Attention.☆127Updated last week
- The official implementation of paper: SimLayerKV: A Simple Framework for Layer-Level KV Cache Reduction.☆44Updated 5 months ago
- ☆37Updated 2 weeks ago
- ☆30Updated 10 months ago
- Pruner-Zero: Evolving Symbolic Pruning Metric from scratch for LLMs☆80Updated 4 months ago
- Activation-aware Singular Value Decomposition for Compressing Large Language Models☆60Updated 5 months ago
- Triton implementation of FlashAttention2 that adds Custom Masks.☆106Updated 7 months ago
- ☆24Updated 4 months ago
- A sparse attention kernel supporting mix sparse patterns☆177Updated last month
- ☆73Updated 2 weeks ago
- ☆52Updated this week
- Fast and memory-efficient exact attention☆67Updated last month
- Implement Flash Attention using Cute.☆74Updated 3 months ago
- Sirius, an efficient correction mechanism, which significantly boosts Contextual Sparsity models on reasoning tasks while maintaining its…☆21Updated 6 months ago
- ☆19Updated 2 weeks ago
- [ICLR 2025] TidalDecode: A Fast and Accurate LLM Decoding with Position Persistent Sparse Attention☆31Updated this week