fkodom / fft-conv-pytorchLinks
Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.
☆516Updated 2 years ago
Alternatives and similar repositories for fft-conv-pytorch
Users that are interested in fft-conv-pytorch are comparing it to the libraries listed below
Sorting:
- DCT (discrete cosine transform) functions for pytorch☆640Updated 3 years ago
- Implementation of Axial attention - attending to multi-dimensional data efficiently☆397Updated 4 years ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆444Updated last year
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆292Updated last year
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆224Updated 10 months ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆641Updated last month
- An All-MLP solution for Vision, from Google AI☆1,056Updated 7 months ago
- Implementation of ConvMixer for "Patches Are All You Need? 🤷"☆1,081Updated 3 years ago
- Differentiable fast wavelet transforms in PyTorch with GPU support.☆413Updated 3 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 9 months ago
- Escaping the Big Data Paradigm with Compact Transformers, 2021 (Train your Vision Transformers in 30 mins on CIFAR-10 with a single GPU!)☆537Updated last year
- Implementation of Linformer for Pytorch☆305Updated 2 years ago
- Learning Rate Warmup in PyTorch☆415Updated 7 months ago
- This is an official pytorch implementation of Fast Fourier Convolution.☆386Updated last year
- An implementation of the efficient attention module.☆328Updated 5 years ago
- 1D interpolation for pytorch☆176Updated last year
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision☆216Updated 4 years ago
- Pytorch implementation of 2D Discrete Wavelet (DWT) and Dual Tree Complex Wavelet Transforms (DTCWT) and a DTCWT based ScatterNet☆1,157Updated 2 years ago
- An implementation of local windowed attention for language modeling☆496Updated 6 months ago
- [ICML 2022, Oral] The PyTorch Implementation of Adaptive Inertia Methods. The algorithms are based on our paper: "Adaptive Inertia: Dise…☆149Updated 2 years ago
- [NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classification☆500Updated 2 years ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆261Updated 4 years ago
- Super Resolution datasets and models in Pytorch☆211Updated 2 years ago
- Adaptive Gradient Clipping☆152Updated 3 years ago
- Estimate/count FLOPS for a given neural network using pytorch☆305Updated 3 years ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆827Updated last year
- A fully invertible U-Net for memory efficiency in Pytorch.☆130Updated 3 years ago
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"☆824Updated 3 years ago
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022☆1,175Updated last year
- Is the attention layer even necessary? (https://arxiv.org/abs/2105.02723)☆485Updated 4 years ago