FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores
☆355Dec 28, 2024Updated last year
Alternatives and similar repositories for flash-fft-conv
Users that are interested in flash-fft-conv are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- FlexAttention w/ FlashAttention3 Support☆27Oct 5, 2024Updated 2 years ago
- Repo for "Monarch Mixer: A Simple Sub-Quadratic GEMM-Based Architecture"☆564Dec 28, 2024Updated last year
- Understand and test language model architectures on synthetic tasks.☆288Mar 22, 2026Updated 6 months ago
- Butterfly matrix multiplication in PyTorch☆182Oct 5, 2023Updated 3 years ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆257Jun 6, 2025Updated last year
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆178Jan 30, 2025Updated last year
- An annotated implementation of the Hyena Hierarchy paper☆34May 28, 2023Updated 3 years ago
- Accelerated First Order Parallel Associative Scan☆198Jan 7, 2026Updated 9 months ago
- ☆45Apr 30, 2018Updated 8 years ago
- Tile primitives for speedy kernels☆3,755Sep 12, 2026Updated last month
- ☆271Jul 11, 2024Updated 2 years ago
- A Quirky Assortment of CuTe Kernels☆1,176Updated this week
- ☆36Apr 12, 2024Updated 2 years ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆605Aug 14, 2026Updated 2 months ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- Convolutions for Sequence Modeling☆920Jun 13, 2024Updated 2 years ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆113Sep 10, 2024Updated 2 years ago
- ☆165Jan 24, 2023Updated 3 years ago
- Language Modeling with the H3 State Space Model☆527Sep 29, 2023Updated 3 years ago
- ☆18Mar 18, 2024Updated 2 years ago
- ☆52Jan 28, 2024Updated 2 years ago
- Some preliminary explorations of Mamba's context scaling.☆223Feb 8, 2024Updated 2 years ago
- Triton-based implementation of Sparse Mixture of Experts.☆280Oct 3, 2025Updated last year
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆77Aug 2, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Official Repository of Pretraining Without Attention (BiGS), BiGS is the first model to achieve BERT-level transfer learning on the GLUE …☆119Mar 16, 2024Updated 2 years ago
- Helpful tools and examples for working with flex-attention☆1,257Updated this week
- Code for the paper: https://arxiv.org/pdf/2309.06979.pdf☆21Jul 29, 2024Updated 2 years ago
- A collection of memory efficient attention operators implemented in the Triton language.☆319Updated this week
- Fast Hadamard transform in CUDA, with a PyTorch interface☆354Mar 10, 2026Updated 7 months ago
- Implementation of https://srush.github.io/annotated-s4☆519Jun 20, 2025Updated last year
- Kernels, of the mega variety :)☆849May 26, 2026Updated 4 months ago
- Accelerating MoE with IO and Tile-aware Optimizations☆780Aug 29, 2026Updated last month
- ☆125May 19, 2025Updated last year
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- Fast Polar Decomposition for Muon☆187Jul 2, 2026Updated 3 months ago
- Official repository for ICML 2024 paper "MoRe Fine-Tuning with 10x Fewer Parameters"☆22Oct 14, 2025Updated last year
- ☆325Jan 8, 2025Updated last year
- NVFP4 Flash-Attention 4 on BlackWell☆68Sep 26, 2026Updated 3 weeks ago
- Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch☆548May 16, 2025Updated last year
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆17Jan 7, 2025Updated last year
- ☆43May 21, 2021Updated 5 years ago