Alternatives and similar repositories for striped_attention:

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

• stanford-futuredata / stk
  ☆104Updated 8 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆88Updated last week
• feifeibear / Odysseus-Transformer
  Odysseus: Playground of LLM Sequence Parallelism
  ☆69Updated 10 months ago
• IST-DASLab / Sparse-Marlin
  Boosting 4-bit inference kernels with 2:4 Sparsity
  ☆73Updated 8 months ago
• Raincleared-Song / sparse_gpu_operator
  GPU operators for sparse tensor operations
  ☆32Updated last year
• Aleph-Alpha / NeurIPS-WANT-submission-efficient-parallelization-layouts
  ☆22Updated last year
• DerrickYLJ / TidalDecode
  [ICLR 2025] TidalDecode: A Fast and Accurate LLM Decoding with Position Persistent Sparse Attention
  ☆35Updated 2 weeks ago
• feifeibear / DPSKV3MFU
  Estimate MFU for DeepSeekV3
  ☆23Updated 4 months ago
• zhuohan123 / terapipe
  ☆72Updated 4 years ago
• INT-FlashAttention2024 / INT-FlashAttention
  ☆68Updated 3 months ago
• RulinShao / LightSeq
  Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training
  ☆209Updated 8 months ago
• Infini-AI-Lab / Sirius
  Sirius, an efficient correction mechanism, which significantly boosts Contextual Sparsity models on reasoning tasks while maintaining its…
  ☆21Updated 7 months ago
• mayank31398 / cute-kernels
  A bunch of kernels that might make stuff slower 😉
  ☆40Updated this week
• andy-yang-1 / DoubleSparse
  16-fold memory access reduction with nearly no loss
  ☆91Updated last month
• microsoft / chunk-attention
  ☆74Updated 2 weeks ago
• RulinShao / FastCkpt
  Python package for rematerialization-aware gradient checkpointing
  ☆24Updated last year
• shawntan / scattermoe
  Triton-based implementation of Sparse Mixture of Experts.
  ☆212Updated 5 months ago
• sustcsonglin / fla-tilelang
  ☆20Updated 2 months ago
• infinigence / FlashOverlap
  A lightweight design for computation-communication overlap.
  ☆67Updated last week
• cchan / tccl
  extensible collectives library in triton
  ☆85Updated last month
• LeiWang1999 / AutoGPTQ.tvm
  GPTQ inference TVM kernel
  ☆38Updated last year
• ankan-ban / llama_cu_awq
  llama INT4 cuda inference with AWQ
  ☆54Updated 3 months ago
• CalvinXKY / mfu_calculation
  A simple calculation for LLM MFU.
  ☆37Updated 2 months ago
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆91Updated 6 years ago
• linxihui / dkernel
  ☆20Updated 2 weeks ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆106Updated 9 months ago
• UofT-EcoSystem / Tempo
  Memory footprint reduction for transformer models
  ☆11Updated 2 years ago
• sail-sg / VocabularyParallelism
  Vocabulary Parallelism
  ☆19Updated last month
• triton-lang / kernels
  ☆78Updated 6 months ago
• PipeFusion / PipeFusion
  A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters
  ☆45Updated 9 months ago