fkodom / fft-conv-pytorch
Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.
☆490Updated last year
Alternatives and similar repositories for fft-conv-pytorch:
Users that are interested in fft-conv-pytorch are comparing it to the libraries listed below
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆559Updated 2 months ago
- Implementation of ConvMixer for "Patches Are All You Need? 🤷"☆1,064Updated 2 years ago
- Implementation of Axial attention - attending to multi-dimensional data efficiently☆371Updated 3 years ago
- An All-MLP solution for Vision, from Google AI☆1,013Updated 5 months ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆271Updated 9 months ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆738Updated 9 months ago
- DCT (discrete cosine transform) functions for pytorch☆592Updated 2 years ago
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision☆217Updated 3 years ago
- Implementation of Linformer for Pytorch☆266Updated last year
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆421Updated 4 months ago
- This is an official pytorch implementation of Fast Fourier Convolution.☆362Updated 8 months ago
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022☆1,088Updated 9 months ago
- [NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classification☆465Updated last year
- An implementation of the efficient attention module.☆301Updated 4 years ago
- Escaping the Big Data Paradigm with Compact Transformers, 2021 (Train your Vision Transformers in 30 mins on CIFAR-10 with a single GPU!)☆511Updated 3 months ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆216Updated last week
- Pytorch implementation of 2D Discrete Wavelet (DWT) and Dual Tree Complex Wavelet Transforms (DTCWT) and a DTCWT based ScatterNet☆1,013Updated last year
- Learning Rate Warmup in PyTorch☆403Updated 2 weeks ago
- Code repository of the paper "Modelling Long Range Dependencies in ND: From Task-Specific to a General Purpose CNN" https://arxiv.org/abs…☆183Updated last year
- Differentiable fast wavelet transforms in PyTorch with GPU support.☆330Updated last week
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"☆809Updated 2 years ago
- An implementation of local windowed attention for language modeling☆420Updated last month
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"☆370Updated last year
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆774Updated 7 months ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,120Updated last year
- [ECCV 2022] Official repository for "MaxViT: Multi-Axis Vision Transformer". SOTA foundation models for classification, detection, segmen…☆457Updated last year
- ☆450Updated last year
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆635Updated 2 months ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆210Updated 2 years ago
- Ranger deep learning optimizer rewrite to use newest components☆327Updated last year