FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores
☆358Dec 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 last year
- Repo for "Monarch Mixer: A Simple Sub-Quadratic GEMM-Based Architecture"☆564Dec 28, 2024Updated last year
- Butterfly matrix multiplication in PyTorch☆180Oct 5, 2023Updated 2 years ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆256Jun 6, 2025Updated last year
- Understand and test language model architectures on synthetic tasks.☆283Mar 22, 2026Updated 4 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆176Jan 30, 2025Updated last year
- Accelerated First Order Parallel Associative Scan☆198Jan 7, 2026Updated 7 months ago
- An annotated implementation of the Hyena Hierarchy paper☆34May 28, 2023Updated 3 years ago
- ☆45Apr 30, 2018Updated 8 years ago
- Tile primitives for speedy kernels☆3,631Jul 13, 2026Updated last month
- A Quirky Assortment of CuTe Kernels☆1,110Aug 8, 2026Updated last week
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆606Updated this week
- ☆271Jul 11, 2024Updated 2 years ago
- Convolutions for Sequence Modeling☆915Jun 13, 2024Updated 2 years ago
- 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.
- ☆35Apr 12, 2024Updated 2 years ago
- ☆165Jan 24, 2023Updated 3 years ago
- ☆18Mar 18, 2024Updated 2 years ago
- Language Modeling with the H3 State Space Model☆526Sep 29, 2023Updated 2 years ago
- ☆52Jan 28, 2024Updated 2 years ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆113Sep 10, 2024Updated last year
- Some preliminary explorations of Mamba's context scaling.☆222Feb 8, 2024Updated 2 years ago
- 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,225Updated this week
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Triton-based implementation of Sparse Mixture of Experts.☆281Oct 3, 2025Updated 10 months ago
- Code for the paper: https://arxiv.org/pdf/2309.06979.pdf☆21Jul 29, 2024Updated 2 years ago
- Implementation of https://srush.github.io/annotated-s4☆519Jun 20, 2025Updated last year
- Accelerating MoE with IO and Tile-aware Optimizations☆741Jul 4, 2026Updated last month
- ☆125May 19, 2025Updated last year
- Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch☆546May 16, 2025Updated last year
- Official repository for ICML 2024 paper "MoRe Fine-Tuning with 10x Fewer Parameters"☆22Oct 14, 2025Updated 10 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆305Updated this week
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆74Aug 2, 2024Updated 2 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆17Jan 7, 2025Updated last year
- Kernels, of the mega variety :)☆804May 26, 2026Updated 2 months ago
- ☆43May 21, 2021Updated 5 years ago
- Fast Hadamard transform in CUDA, with a PyTorch interface☆345Mar 10, 2026Updated 5 months ago
- Triton Implementation of HyperAttention Algorithm☆48Dec 11, 2023Updated 2 years ago
- HGRN2: Gated Linear RNNs with State Expansion☆58Aug 20, 2024Updated last year
- Building blocks for foundation models.☆639Jan 3, 2024Updated 2 years ago