Doraemonzzz / Awesome-Triton-Resources

Related projects ⓘ

Alternatives and complementary repositories for Awesome-Triton-Resources

• Aleph-Alpha / NeurIPS-WANT-submission-efficient-parallelization-layouts
  ☆22Updated 11 months ago
• emalach / LinearLM
  Code for the paper: https://arxiv.org/pdf/2309.06979.pdf
  ☆16Updated 3 months ago
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated last month
• sustcsonglin / mamba-triton
  ☆45Updated 9 months ago
• ahennequ / pytorch-custom-mma
  ☆29Updated 2 years ago
• srush / tangent
  Source-to-Source Debuggable Derivatives in Pure Python
  ☆14Updated 10 months ago
• BlinkDL / LinearAttentionArena
  Here we will test various linear attention designs.
  ☆56Updated 6 months ago
• renll / SeqBoat
  [NeurIPS 2023] Sparse Modular Activation for Efficient Sequence Modeling
  ☆35Updated 11 months ago
• HazyResearch / prefix-linear-attention
  ☆45Updated 4 months ago
• yikangshen / megablocks
  ☆18Updated 5 months ago
• sjelassi / transformers_ssm_copy
  ☆24Updated 8 months ago
• proger / nanokitchen
  Parallel Associative Scan for Language Models
  ☆18Updated 10 months ago
• Edward-Sun / gpt-accelera
  Simple and efficient pytorch-native transformer training and inference (batched)
  ☆61Updated 7 months ago
• Doraemonzzz / hgru2-pytorch
  ☆21Updated last month
• amirzandieh / HyperAttention
  Triton Implementation of HyperAttention Algorithm
  ☆46Updated 11 months ago
• shreyansh26 / Attention-Mask-Patterns
  Using FlexAttention to compute attention with different masking patterns
  ☆40Updated 2 months ago
• BBuf / flash-rwkv
  ☆29Updated 5 months ago
• srush / LLM-Talk
  ☆47Updated 9 months ago
• dame-cell / Triformer
  Transformers components but in Triton
  ☆27Updated this week
• softmax1 / Flash-Attention-Softmax-N
  CUDA and Triton implementations of Flash Attention with SoftmaxN.
  ☆66Updated 5 months ago
• shawntan / stickbreaking-attention
  Stick-breaking attention
  ☆34Updated 2 weeks ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆16Updated 4 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆72Updated 10 months ago
• Ryu1845 / hyena-jax
  Implementation of Hyena Hierarchy in JAX
  ☆10Updated last year
• dangxingyu / rnn-icrag
  Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"
  ☆24Updated 7 months ago
• habanero-lab / APPy
  APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…
  ☆20Updated last week
• glassroom / heinsen_attention
  Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)
  ☆24Updated 5 months ago
• srush / mamba-scans
  Blog post
  ☆16Updated 9 months ago
• Doraemonzzz / nanoTransNormer
  ☆11Updated last year