fkodom / fft-conv-pytorchLinks
Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.
β499Updated 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
Sorting:
- Implementation of ConvMixer for "Patches Are All You Need? π€·"β1,074Updated 2 years ago
- An All-MLP solution for Vision, from Google AIβ1,025Updated 9 months ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)β218Updated 3 months ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.β430Updated 8 months ago
- DCT (discrete cosine transform) functions for pytorchβ614Updated 2 years ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch modelβ592Updated 6 months ago
- This is an official pytorch implementation of Fast Fourier Convolution.β373Updated last year
- Escaping the Big Data Paradigm with Compact Transformers, 2021 (Train your Vision Transformers in 30 mins on CIFAR-10 with a single GPU!)β534Updated 7 months ago
- [NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classificationβ479Updated 2 years ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence lengthβ775Updated last year
- Implementation of a U-net complete with efficient attention as well as the latest research findingsβ282Updated last year
- Code repository of the paper "Modelling Long Range Dependencies in ND: From Task-Specific to a General Purpose CNN" https://arxiv.org/absβ¦β184Updated last month
- Implementation of fused cosine similarity attention in the same style as Flash Attentionβ214Updated 2 years ago
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Visionβ217Updated 4 years ago
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022β1,122Updated last year
- Pytorch implementation of 2D Discrete Wavelet (DWT) and Dual Tree Complex Wavelet Transforms (DTCWT) and a DTCWT based ScatterNetβ1,081Updated last year
- An implementation of the efficient attention module.β317Updated 4 years ago
- Implementation of Linformer for Pytorchβ288Updated last year
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(nΒ²) Memory"β379Updated last year
- Differentiable fast wavelet transforms in PyTorch with GPU support.β363Updated last month
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"β817Updated 2 years ago
- An implementation of local windowed attention for language modelingβ454Updated 5 months ago
- Learning Rate Warmup in PyTorchβ410Updated this week
- Implementation of Axial attention - attending to multi-dimensional data efficientlyβ384Updated 3 years ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transformsβ260Updated 4 years ago
- Estimate/count FLOPS for a given neural network using pytorchβ304Updated 3 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilitiesβ209Updated last year
- [ICLR 2023] "More ConvNets in the 2020s: Scaling up Kernels Beyond 51x51 using Sparsity"; [ICML 2023] "Are Large Kernels Better Teachersβ¦β274Updated last year
- Unofficial Implementation of MLP-Mixer, gMLP, resMLP, Vision Permutator, S2MLP, S2MLPv2, RaftMLP, HireMLP, ConvMLP, AS-MLP, SparseMLP, Coβ¦β169Updated 2 years ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.neβ¦β116Updated 2 years ago