Alternatives and similar repositories for Flash-Attention-with-Bias-Triton

Users that are interested in Flash-Attention-with-Bias-Triton are comparing it to the libraries listed below

• shawntan / stickbreaking-attention
  Stick-breaking attention
  ☆62Updated 6 months ago
• liusulin / DDPD
  Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]
  ☆84Updated 8 months ago
• HKUNLP / reparam-discrete-diffusion
  Reparameterized Discrete Diffusion Models for Text Generation
  ☆104Updated 2 years ago
• google-deepmind / md4
  Official Jax Implementation of MD4 Masked Diffusion Models
  ☆151Updated 10 months ago
• igul222 / plaid
  ☆111Updated 2 years ago
• HKUNLP / diffusion-vs-ar
  [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"
  ☆87Updated 10 months ago
• insuhan / hyper-attn
  ☆83Updated 2 years ago
• dvruette / gidd
  Code accompanying the paper "Generalized Interpolating Discrete Diffusion"
  ☆112Updated 7 months ago
• Doraemonzzz / Awesome-Triton-Resources
  Awesome Triton Resources
  ☆39Updated 8 months ago
• xhan77 / ssd-lm
  Semi-autoregressive Simplex-based Diffusion Language Model for Text Generation and Modular Control
  ☆76Updated 3 years ago
• MinkaiXu / Energy-Diffusion-LLM
  Reproduce ICLR2025 Energy-Based Diffusion Language Models for Text Generation
  ☆52Updated 5 months ago
• Aleph-Alpha-Research / NeurIPS-WANT-submission-efficient-parallelization-layouts
  ☆22Updated 2 years ago
• ruqizhang / discrete-langevin
  ☆42Updated 3 years ago
• alexzhang13 / flashattention2-custom-mask
  Triton implementation of FlashAttention2 that adds Custom Masks.
  ☆159Updated last year
• gregorbachmann / Next-Token-Failures
  ☆107Updated last year
• OpenNLPLab / HGRN
  [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…
  ☆66Updated last year
• thu-ml / Noise-Contrastive-Alignment
  Code accompanying the paper "Noise Contrastive Alignment of Language Models with Explicit Rewards" (NeurIPS 2024)
  ☆58Updated last year
• HazyResearch / prefix-linear-attention
  ☆57Updated last year
• ozekri / SEPO
  Code for the paper: "Fine-Tuning Discrete Diffusion Models with Policy Gradient Methods"
  ☆31Updated 7 months ago
• fla-org / fla-zoo
  Flash-Linear-Attention models beyond language
  ☆20Updated 4 months ago
• andrew-cr / discrete_flow_models
  Code for the paper https://arxiv.org/abs/2402.04997
  ☆102Updated last year
• sustcsonglin / linear-attention-and-beyond-slides
  ☆102Updated 10 months ago
• sjelassi / transformers_ssm_copy
  ☆35Updated last year
• gl-ybnbxb / BoNBoN
  ☆18Updated last year
• zhixuan-lin / forgetting-transformer
  [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning
  ☆134Updated 3 weeks ago
• glassroom / heinsen_attention
  Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)
  ☆24Updated last year
• bansky-cl / diffusion-nlp-paper-arxiv
  Auto get diffusion nlp papers in Axriv. More papers Information can be found in another repository "Diffusion-LM-Papers".
  ☆248Updated this week
• lebellig / discrete-fm
  Educational implementation of the Discrete Flow Matching paper
  ☆127Updated last year
• 00ffcc / chunkRWKV6
  continous batching and parallel acceleration for RWKV6
  ☆22Updated last year
• xiayuqing0622 / flex_head_fa
  Fast and memory-efficient exact attention
  ☆75Updated 10 months ago