NX-AI / flashrnnLinks
FlashRNN - Fast RNN Kernels with I/O Awareness
☆92Updated last month
Alternatives and similar repositories for flashrnn
Users that are interested in flashrnn are comparing it to the libraries listed below
Sorting:
- Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.☆64Updated 2 weeks ago
- Accelerated First Order Parallel Associative Scan☆182Updated 10 months ago
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆185Updated 3 months ago
- The evaluation framework for training-free sparse attention in LLMs☆82Updated 3 weeks ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆128Updated 7 months ago
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆138Updated last month
- ☆222Updated last month
- Work in progress.☆70Updated 2 weeks ago
- Normalized Transformer (nGPT)☆184Updated 7 months ago
- Fast and memory-efficient exact attention☆68Updated 4 months ago
- ☆37Updated 3 months ago
- Load compute kernels from the Hub☆203Updated this week
- ☆116Updated last month
- ☆51Updated last year
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆127Updated 10 months ago
- Experiment of using Tangent to autodiff triton☆79Updated last year
- DPO, but faster 🚀☆43Updated 7 months ago
- RWKV is an RNN with transformer-level LLM performance. It can be directly trained like a GPT (parallelizable). So it's combining the best…☆49Updated 3 months ago
- Awesome Triton Resources☆31Updated 2 months ago
- ring-attention experiments☆144Updated 8 months ago
- RWKV-7: Surpassing GPT☆92Updated 7 months ago
- Implementations of attention with the softpick function, naive and FlashAttention-2☆80Updated 2 months ago
- This repository contains code for the MicroAdam paper.☆20Updated 7 months ago
- JAX bindings for Flash Attention v2☆90Updated last year
- ☆71Updated 2 weeks ago
- ☆82Updated 10 months ago
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆78Updated last month
- This repository contains the experimental PyTorch native float8 training UX☆224Updated 11 months ago
- Griffin MQA + Hawk Linear RNN Hybrid☆87Updated last year
- FlexAttention w/ FlashAttention3 Support☆26Updated 9 months ago