Alternatives and similar repositories for flash-stu

Users that are interested in flash-stu are comparing it to the libraries listed below

• TiledTensor / TiledBench

  View on GitHub
  Benchmark tests supporting the TiledCUDA library.
  ☆18Nov 19, 2024Updated last year
• ACA-Lab-SJTU / token-ring

  View on GitHub
  ☆13Jan 7, 2025Updated last year
• deep-spin / sparse-communication

  View on GitHub
  ☆12Mar 7, 2022Updated 3 years ago
• aws-neuron / nki-library

  View on GitHub
  ☆42Jan 24, 2026Updated 3 weeks ago
• google-deepmind / spectral_ssm

  View on GitHub
  ☆35Apr 12, 2024Updated last year
• sustcsonglin / linear-attention-and-beyond-slides

  View on GitHub
  ☆105Feb 25, 2025Updated 11 months ago
• belindal / state-tracking

  View on GitHub
  Code and data for paper "(How) do Language Models Track State?"
  ☆21Mar 31, 2025Updated 10 months ago
• cassiewilliam / cuda_op_benchmark

  View on GitHub
  方便扩展的Cuda算子理解和优化框架，仅用在学习使用
  ☆18Jun 13, 2024Updated last year
• radarFudan / Curse-of-memory

  View on GitHub
  Curse-of-memory phenomenon of RNNs in sequence modelling
  ☆19May 8, 2025Updated 9 months ago
• Dao-AILab / gemm-cublas

  View on GitHub
  ☆22May 5, 2025Updated 9 months ago
• HanGuo97 / hilt

  View on GitHub
  ☆39Dec 14, 2025Updated 2 months ago
• NonvolatileMemory / flash_tree_attn

  View on GitHub
  ☆20Dec 24, 2024Updated last year
• yester31 / Cutlass_EX

  View on GitHub
  study of cutlass
  ☆22Nov 10, 2024Updated last year
• NathanGodey / headless-lm

  View on GitHub
  Training and evaluation code for the paper "Headless Language Models: Learning without Predicting with Contrastive Weight Tying" (https:/…
  ☆28Apr 17, 2024Updated last year
• muriloboratto / NVSHEMEM

  View on GitHub
  Sample Codes using NVSHMEM on Multi-GPU
  ☆30Jan 22, 2023Updated 3 years ago
• JohndeVostok / APE

  View on GitHub
  A GPU FP32 computation method with Tensor Cores.
  ☆26Dec 8, 2025Updated 2 months ago
• ambisinister / mla-experiments

  View on GitHub
  Experiments on Multi-Head Latent Attention
  ☆99Aug 19, 2024Updated last year
• glassroom / heinsen_attention

  View on GitHub
  Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)
  ☆24Jun 6, 2024Updated last year
• jopetty / word-problem

  View on GitHub
  Experiments on the impact of depth in transformers and SSMs.
  ☆40Oct 23, 2025Updated 3 months ago
• KuangjuX / AttnLink

  View on GitHub
  An experimental communicating attention kernel based on DeepEP.
  ☆35Jul 29, 2025Updated 6 months ago
• ahennequ / pytorch-custom-mma

  View on GitHub
  ☆29Oct 3, 2022Updated 3 years ago
• AndPotap / einsum-search

  View on GitHub
  ☆33Oct 4, 2024Updated last year
• tk-rusch / unicornn

  View on GitHub
  Official code for UnICORNN (ICML 2021)
  ☆28Oct 1, 2021Updated 4 years ago
• kazuki-irie / kv-memory-brain

  View on GitHub
  Official Code Repository for the paper "Key-value memory in the brain"
  ☆31Feb 25, 2025Updated 11 months ago
• GindaChen / FlexFlashAttention3

  View on GitHub
  FlexAttention w/ FlashAttention3 Support
  ☆27Oct 5, 2024Updated last year
• ademeure / DeeperGEMM

  View on GitHub
  DeeperGEMM: crazy optimized version
  ☆73May 5, 2025Updated 9 months ago
• shawnricecake / draft-attention

  View on GitHub
  Code for Draft Attention
  ☆99May 22, 2025Updated 8 months ago
• NX-AI / flashrnn

  View on GitHub
  FlashRNN - Fast RNN Kernels with I/O Awareness
  ☆174Oct 20, 2025Updated 3 months ago
• proger / accelerated-scan

  View on GitHub
  Accelerated First Order Parallel Associative Scan
  ☆196Jan 7, 2026Updated last month
• Doraemonzzz / Awesome-Triton-Resources

  View on GitHub
  Awesome Triton Resources
  ☆39Apr 27, 2025Updated 9 months ago
• gty111 / GEMM_MMA

  View on GitHub
  Optimize GEMM with tensorcore step by step
  ☆36Dec 17, 2023Updated 2 years ago
• NYCU-AI-EDA / Netlistify

  View on GitHub
  ☆26Dec 3, 2025Updated 2 months ago
• PiotrNawrot / sparse-frontier

  View on GitHub
  The evaluation framework for training-free sparse attention in LLMs
  ☆119Jan 27, 2026Updated 2 weeks ago
• tilde-research / nsa-impl

  View on GitHub
  An efficient implementation of the NSA (Native Sparse Attention) kernel
  ☆129Jun 24, 2025Updated 7 months ago
• catqaq / NLP-Notes

  View on GitHub
  详细双语注释版word2vec源码，well-annotated word2vec
  ☆10Oct 3, 2021Updated 4 years ago
• madsys-dev / deepseekv2-profile

  View on GitHub
  ☆155Mar 4, 2025Updated 11 months ago
• gty111 / PTX-EMU

  View on GitHub
  PTX-EMU is a simple emulator for CUDA program.
  ☆37Apr 25, 2025Updated 9 months ago
• nanomaoli / llm_reproducibility

  View on GitHub
  ☆77Updated this week
• luliyucoordinate / cute-flash-attention

  View on GitHub
  Implement Flash Attention using Cute.
  ☆100Dec 17, 2024Updated last year