OpenBMB / infllmv2_cuda_implLinks
☆28Updated last month
Alternatives and similar repositories for infllmv2_cuda_impl
Users that are interested in infllmv2_cuda_impl are comparing it to the libraries listed below
Sorting:
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆215Updated last month
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆38Updated last month
- Odysseus: Playground of LLM Sequence Parallelism☆70Updated last year
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆119Updated last month
- ☆119Updated last month
- ☆90Updated 3 months ago
- ☆75Updated last month
- [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring☆200Updated last week
- A sparse attention kernel supporting mix sparse patterns☆256Updated 5 months ago
- ☆60Updated 2 months ago
- Quantized Attention on GPU☆44Updated 7 months ago
- Accelerate LLM preference tuning via prefix sharing with a single line of code☆42Updated last week
- QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.☆135Updated 3 months ago
- qwen-nsa☆68Updated 3 months ago
- ☆49Updated last month
- A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters☆47Updated 11 months ago
- ☆77Updated 2 months ago
- ☆96Updated 10 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆101Updated last month
- Efficient triton implementation of Native Sparse Attention.☆176Updated last month
- 16-fold memory access reduction with nearly no loss☆100Updated 3 months ago
- Triton implementation of Flash Attention2.0☆36Updated last year
- Vocabulary Parallelism☆19Updated 4 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆272Updated last year
- An easy-to-use package for implementing SmoothQuant for LLMs☆102Updated 3 months ago
- A parallelism VAE avoids OOM for high resolution image generation☆67Updated 5 months ago
- Implement Flash Attention using Cute.☆88Updated 7 months ago
- ☆50Updated 2 months ago
- A curated list of recent papers on efficient video attention for video diffusion models, including sparsification, quantization, and cach…☆28Updated this week
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆165Updated last year