Alternatives and similar repositories for FlashMHA

Users that are interested in FlashMHA are comparing it to the libraries listed below

• li-plus / flash-preference
  Accelerate LLM preference tuning via prefix sharing with a single line of code
  ☆41Updated last month
• OpenNLPLab / LASP
  Linear Attention Sequence Parallelism (LASP)
  ☆85Updated last year
• VITA-Group / Ms-PoE
  "Found in the Middle: How Language Models Use Long Contexts Better via Plug-and-Play Positional Encoding" Zhenyu Zhang, Runjin Chen, Shiw…
  ☆29Updated last year
• rayleizhu / vllm-ra
  [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts
  ☆40Updated last year
• BBuf / flash-rwkv
  ☆31Updated last year
• GeeeekExplorer / transformers-patch
  patches for huggingface transformers to save memory
  ☆24Updated 3 weeks ago
• tilde-research / nsa-impl
  An efficient implementation of the NSA (Native Sparse Attention) kernel
  ☆72Updated this week
• pprp / Pruner-Zero
  [ICML24] Pruner-Zero: Evolving Symbolic Pruning Metric from scratch for LLMs
  ☆84Updated 7 months ago
• Infini-AI-Lab / gsm_infinite
  ☆47Updated 2 weeks ago
• tridao / flash-attention-wheels
  ☆50Updated last year
• Infini-AI-Lab / GRESO
  ☆36Updated last week
• yuzhenmao / IceFormer
  Implementation of IceFormer: Accelerated Inference with Long-Sequence Transformers on CPUs (ICLR 2024).
  ☆25Updated last year
• microsoft / AttentionEngine
  ☆71Updated last month
• xiayuqing0622 / flex_head_fa
  Fast and memory-efficient exact attention
  ☆68Updated 3 months ago
• feifeibear / Odysseus-Transformer
  Odysseus: Playground of LLM Sequence Parallelism
  ☆70Updated last year
• keshik6 / grafting
  Exploring Diffusion Transformer Designs via Grafting
  ☆41Updated last week
• shreyansh26 / Attention-Mask-Patterns
  Using FlexAttention to compute attention with different masking patterns
  ☆44Updated 9 months ago
• Dao-AILab / grouped-latent-attention
  ☆114Updated 3 weeks ago
• feifeibear / ChituAttention
  Quantized Attention on GPU
  ☆44Updated 7 months ago
• PipeFusion / PipeFusion
  A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters
  ☆47Updated 11 months ago
• Infini-AI-Lab / Multiverse
  ☆58Updated last week
• TianjinYellow / StableSPAM
  ☆21Updated 2 months ago
• LiqunMa / FBI-LLM
  FBI-LLM: Scaling Up Fully Binarized LLMs from Scratch via Autoregressive Distillation
  ☆49Updated 11 months ago
• SprocketLab / sparse_matrix_fine_tuning
  Official repository for ICML 2024 paper "MoRe Fine-Tuning with 10x Fewer Parameters"
  ☆20Updated last month
• kyegomez / Hedgehog
  Implementation of the model "Hedgehog" from the paper: "The Hedgehog & the Porcupine: Expressive Linear Attentions with Softmax Mimicry"
  ☆14Updated last year
• lfsszd / CS-Drafting
  Cascade Speculative Drafting
  ☆29Updated last year
• fla-org / flash-bidirectional-linear-attention
  Triton implement of bi-directional (non-causal) linear attention
  ☆50Updated 4 months ago
• sustcsonglin / linear-attention-and-beyond-slides
  ☆76Updated 4 months ago
• metacarbon / shareAtt
  Beyond KV Caching: Shared Attention for Efficient LLMs
  ☆19Updated 11 months ago
• RWKV / RWKV-LM
  RWKV is an RNN with transformer-level LLM performance. It can be directly trained like a GPT (parallelizable). So it's combining the best…
  ☆47Updated 3 months ago