fkodom / fft-conv-pytorch
Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.
β475Updated last year
Related projects β
Alternatives and complementary repositories for fft-conv-pytorch
- This is an official pytorch implementation of Fast Fourier Convolution.β345Updated 5 months ago
- Implementation of ConvMixer for "Patches Are All You Need? π€·"β1,062Updated 2 years ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch modelβ514Updated 3 weeks ago
- Pytorch implementation of 2D Discrete Wavelet (DWT) and Dual Tree Complex Wavelet Transforms (DTCWT) and a DTCWT based ScatterNetβ969Updated last year
- Implementation of a U-net complete with efficient attention as well as the latest research findingsβ264Updated 6 months ago
- [NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classificationβ445Updated 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!)β499Updated last week
- Neighborhood Attention Extension. Bringing attention to a neighborhood near you!β362Updated last week
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Visionβ209Updated 3 years ago
- An All-MLP solution for Vision, from Google AIβ1,001Updated 2 months 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β696Updated 6 months ago
- Implementation of Axial attention - attending to multi-dimensional data efficientlyβ352Updated 3 years 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
- 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
- DCT (discrete cosine transform) functions for pytorchβ571Updated 2 years ago
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Modelsβ759Updated 4 months ago
- Implementation of Linformer for Pytorchβ255Updated 10 months ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.β405Updated last month
- (N=1,2,3)-dimensional unfold (im2col) and fold (col2im) in PyTorchβ85Updated 4 months ago
- Learning Rate Warmup in PyTorchβ393Updated last week
- Implementation of fused cosine similarity attention in the same style as Flash Attentionβ208Updated last year
- An implementation of the efficient attention module.β283Updated 3 years ago
- A library for calculating the FLOPs in the forward() process based on torch.fxβ80Updated 2 months ago
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022β1,053Updated 5 months ago
- Ranger deep learning optimizer rewrite to use newest componentsβ322Updated 8 months ago
- LeViT a Vision Transformer in ConvNet's Clothing for Faster Inferenceβ602Updated 2 years ago
- [ICLR 2021 top 3%] Is Attention Better Than Matrix Decomposition?β325Updated 2 years ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.neβ¦β115Updated last year
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"β806Updated 2 years ago