HazyResearch / flash-fft-convLinks
FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores
☆325Updated 7 months ago
Alternatives and similar repositories for flash-fft-conv
Users that are interested in flash-fft-conv are comparing it to the libraries listed below
Sorting:
- Accelerated First Order Parallel Associative Scan☆184Updated 11 months ago
- This repository contains the experimental PyTorch native float8 training UX☆224Updated last year
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆565Updated this week
- The AdEMAMix Optimizer: Better, Faster, Older.☆184Updated 10 months ago
- A library for unit scaling in PyTorch☆128Updated 3 weeks ago
- Annotated version of the Mamba paper☆487Updated last year
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI☆289Updated 2 months ago
- Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch☆533Updated 2 months ago
- ☆323Updated this week
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆274Updated 2 weeks ago
- A repository for log-time feedforward networks☆222Updated last year
- Fast Hadamard transform in CUDA, with a PyTorch interface☆213Updated last year
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"☆379Updated 2 years ago
- Helpful tools and examples for working with flex-attention☆908Updated 2 weeks ago
- Collection of kernels written in Triton language☆142Updated 4 months ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆238Updated 2 months ago
- ☆293Updated 3 months ago
- Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793☆431Updated 2 months ago
- ☆147Updated 2 years ago
- FlashRNN - Fast RNN Kernels with I/O Awareness☆93Updated last month
- ☆162Updated last year
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆112Updated 8 months ago
- ☆206Updated 8 months ago
- Efficient optimizers☆252Updated last week
- Explorations into the recently proposed Taylor Series Linear Attention☆100Updated 11 months ago
- JAX bindings for Flash Attention v2☆91Updated last week
- Fast low-bit matmul kernels in Triton☆338Updated last week
- Implementation of Flash Attention in Jax☆215Updated last year
- Scalable and Performant Data Loading☆291Updated this week
- ☆158Updated last year