☆46Nov 10, 2023Updated 2 years ago
Alternatives and similar repositories for striped_attention
Users that are interested in striped_attention are comparing it to the libraries listed below
Sorting:
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆222Aug 19, 2024Updated last year
- Instructions and templates for SC authors☆17Aug 22, 2021Updated 4 years ago
- ☆20Dec 24, 2024Updated last year
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆24Jun 6, 2024Updated last year
- Distributed SDDMM Kernel☆12Jul 8, 2022Updated 3 years ago
- An optimized Merkle Patricia Trie implementation on GPU, fully compatible with and integrable into Ethereum. The paper is published on VL…☆14Apr 15, 2024Updated last year
- Large Context Attention☆769Oct 13, 2025Updated 4 months ago
- Ring attention implementation with flash attention☆987Sep 10, 2025Updated 5 months ago
- ☆28Jan 12, 2022Updated 4 years ago
- ☆32May 26, 2024Updated last year
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆161Oct 13, 2025Updated 4 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆644Jan 15, 2026Updated last month
- [COLM 2024] TriForce: Lossless Acceleration of Long Sequence Generation with Hierarchical Speculative Decoding☆277Aug 31, 2024Updated last year
- A low-latency & high-throughput serving engine for LLMs☆482Jan 8, 2026Updated 2 months ago
- ☆39Apr 5, 2024Updated last year
- ☆17Feb 19, 2024Updated 2 years ago
- Compiler with automatic differentiation☆49Oct 18, 2023Updated 2 years ago
- ☆23Aug 21, 2025Updated 6 months ago
- A repository of freely available SVG drawings for your publications, presentations etc.☆17Nov 29, 2015Updated 10 years ago
- Pytorch process group third-party plugin for UCC☆21Apr 15, 2024Updated last year
- Code for the paper: https://arxiv.org/pdf/2309.06979.pdf☆21Jul 29, 2024Updated last year
- Xmixers: A collection of SOTA efficient token/channel mixers☆28Sep 4, 2025Updated 6 months ago
- Linear Attention Sequence Parallelism (LASP)☆89Jun 4, 2024Updated last year
- Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.☆91Nov 23, 2022Updated 3 years ago
- ☆50Jun 16, 2025Updated 8 months ago
- ☆33Oct 4, 2024Updated last year
- ☆30Jul 22, 2024Updated last year
- Train, tune, and infer Bamba model☆137Jun 4, 2025Updated 9 months ago
- ☆26Mar 14, 2024Updated last year
- Official Code Repository for the paper "Key-value memory in the brain"☆31Feb 25, 2025Updated last year
- A framework for pipelined computing on GPU☆30Jul 17, 2019Updated 6 years ago
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆478Mar 15, 2024Updated last year
- [NeurIPS'23] H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Models.☆504Aug 1, 2024Updated last year
- ☆27Oct 26, 2019Updated 6 years ago
- Ok-Topk is a scheme for distributed training with sparse gradients. Ok-Topk integrates a novel sparse allreduce algorithm (less than 6k c…☆27Dec 10, 2022Updated 3 years ago
- Awesome Triton Resources☆39Apr 27, 2025Updated 10 months ago
- A pattern-based algorithmic autotuner for graph processing on GPUs.☆32Jun 25, 2025Updated 8 months ago
- Optimize GEMM with tensorcore step by step☆36Dec 17, 2023Updated 2 years ago
- Transformer Grammars: Augmenting Transformer Language Models with Syntactic Inductive Biases at Scale, TACL (2022)☆133Feb 20, 2026Updated 2 weeks ago