maximzubkov / fft-scan
Efficient PScan implementation in PyTorch
☆15Updated last year
Alternatives and similar repositories for fft-scan:
Users that are interested in fft-scan are comparing it to the libraries listed below
- Blog post☆16Updated 11 months ago
- Parallel Associative Scan for Language Models☆18Updated last year
- ☆51Updated 8 months ago
- ☆46Updated last year
- ☆30Updated 11 months ago
- Triton Implementation of HyperAttention Algorithm☆46Updated last year
- ☆30Updated 2 months ago
- Experiments on the impact of depth in transformers and SSMs.☆22Updated 2 months ago
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆62Updated 9 months ago
- ☆11Updated last year
- ☆29Updated 3 months ago
- ☆49Updated 6 months ago
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated 8 months ago
- ☆19Updated 4 months ago
- Jax implementation of "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"☆13Updated 8 months ago
- ☆53Updated last year
- Stick-breaking attention☆41Updated 2 weeks ago
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆25Updated 9 months ago
- ☆37Updated 9 months ago
- ☆50Updated 3 months ago
- ☆31Updated 9 months ago
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆24Updated 7 months ago
- Here we will test various linear attention designs.☆58Updated 9 months ago
- Using FlexAttention to compute attention with different masking patterns☆40Updated 4 months ago
- ☆32Updated last year
- Source-to-Source Debuggable Derivatives in Pure Python☆15Updated last year
- Code for the paper: https://arxiv.org/pdf/2309.06979.pdf☆18Updated 6 months ago
- [NeurIPS 2023] Sparse Modular Activation for Efficient Sequence Modeling☆35Updated last year
- Experiment of using Tangent to autodiff triton☆74Updated last year
- ☆29Updated 2 years ago