fkodom / fft-conv-pytorch
Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.
☆477Updated last year
Related projects ⓘ
Alternatives and complementary repositories for fft-conv-pytorch
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆517Updated last month
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆406Updated last month
- Implementation of ConvMixer for "Patches Are All You Need? 🤷"☆1,062Updated 2 years ago
- An All-MLP solution for Vision, from Google AI☆1,003Updated 2 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!)☆500Updated 2 weeks ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆208Updated this week
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆698Updated 6 months ago
- Learning Rate Warmup in PyTorch☆392Updated this week
- Neighborhood Attention Extension. Bringing attention to a neighborhood near you!☆366Updated this week
- DCT (discrete cosine transform) functions for pytorch☆572Updated 2 years ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆265Updated 6 months ago
- Estimate/count FLOPS for a given neural network using pytorch☆304Updated 2 years ago
- [NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classification☆445Updated last year
- A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.☆119Updated 11 months ago
- Code repository of the paper "Modelling Long Range Dependencies in ND: From Task-Specific to a General Purpose CNN" https://arxiv.org/abs…☆181Updated last year
- This is an official pytorch implementation of Fast Fourier Convolution.☆347Updated 5 months ago
- Implementation of Linformer for Pytorch☆257Updated 10 months ago
- Implementation of Axial attention - attending to multi-dimensional data efficiently☆352Updated 3 years ago
- An implementation of local windowed attention for language modeling☆384Updated 2 months ago
- A general and accurate MACs / FLOPs profiler for PyTorch models☆571Updated 6 months ago
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision☆209Updated 3 years ago
- Scaling Up Your Kernels to 31x31: Revisiting Large Kernel Design in CNNs (CVPR 2022)☆870Updated 6 months ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆207Updated last year
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022☆1,054Updated 6 months ago
- ☆566Updated 3 weeks ago
- Pytorch implementation of 2D Discrete Wavelet (DWT) and Dual Tree Complex Wavelet Transforms (DTCWT) and a DTCWT based ScatterNet☆977Updated last year
- The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).☆548Updated last year
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆760Updated 4 months ago
- Compare neural networks by their feature similarity☆343Updated last year
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆140Updated 3 years ago