Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.
☆517Sep 28, 2023Updated 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:
- This is an official pytorch implementation of Fast Fourier Convolution.☆385May 25, 2024Updated last year
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆141Dec 21, 2021Updated 4 years ago
- DCT (discrete cosine transform) functions for pytorch☆643Nov 6, 2022Updated 3 years ago
- Code for the paper "Unsupervised Contrastive Learning of Sound Event Representations", ICASSP 2021.☆93Dec 22, 2022Updated 3 years ago
- Implementation for NATv2.☆23Feb 20, 2021Updated 5 years ago
- Fast PyTorch based DSP for audio and 1D signals☆452Feb 17, 2025Updated last year
- Udacity CS344 Introduction to Parallell Programming (https://classroom.udacity.com/courses/cs344), with assignments/materials updated to …☆45Sep 16, 2021Updated 4 years ago
- Pytorch implementation of 2D Discrete Wavelet (DWT) and Dual Tree Complex Wavelet Transforms (DTCWT) and a DTCWT based ScatterNet☆1,158Aug 2, 2023Updated 2 years ago
- Accompanying code for our paper "Optimizing Short-Time Fourier Transform Parameters via Gradient Descent"☆34Oct 30, 2020Updated 5 years ago
- Fast, differentiable sorting and ranking in PyTorch☆850Jun 9, 2025Updated 8 months ago
- A temporal module for PyTorch-ComplexTensor☆44Jun 28, 2024Updated last year
- The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).☆563Jul 13, 2023Updated 2 years ago
- ☆41Apr 3, 2021Updated 4 years ago
- Relative Positional Encoding for Transformers with Linear Complexity☆65Apr 1, 2022Updated 3 years ago
- Audio processing by using pytorch 1D convolution network☆1,117Dec 7, 2025Updated 2 months ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,163Mar 22, 2024Updated last year
- Cyclic Differentiable Architecture Search☆36Feb 14, 2022Updated 4 years ago
- Pytorch implementation of subband decomposition☆92Jul 26, 2022Updated 3 years ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Dec 4, 2021Updated 4 years ago
- Source code for "On the Relationship between Self-Attention and Convolutional Layers"☆1,116Jan 10, 2023Updated 3 years ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,415Feb 20, 2026Updated 2 weeks ago
- [NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classification☆505Jun 12, 2023Updated 2 years ago
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022☆1,175May 15, 2024Updated last year
- Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains☆1,361Jan 17, 2023Updated 3 years ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,471May 2, 2025Updated 10 months ago
- Is the attention layer even necessary? (https://arxiv.org/abs/2105.02723)☆485May 7, 2021Updated 4 years ago
- pytorch implementation of complex convolutional neural network☆172May 15, 2020Updated 5 years ago
- Pytorch library for fast transformer implementations☆1,763Mar 23, 2023Updated 2 years ago
- Count the MACs / FLOPs of your PyTorch model.☆5,082Jul 8, 2024Updated last year
- High-fidelity performance metrics for generative models in PyTorch☆1,174Feb 17, 2026Updated 2 weeks ago
- ☆53May 15, 2025Updated 9 months ago
- Pytorch implementation of the invertible CQT based on Non-stationary Gabor filters☆36Jun 20, 2023Updated 2 years ago
- ☆249Mar 16, 2022Updated 3 years ago
- Code release for ConvNeXt model☆6,302Jan 8, 2023Updated 3 years ago
- Collection of audio-focused loss functions in PyTorch☆854Jul 30, 2024Updated last year
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,225Feb 25, 2026Updated last week
- ☆508Jun 25, 2024Updated last year
- Faster depthwise convolutions for PyTorch☆66Feb 20, 2020Updated 6 years ago
- LEAF is a learnable alternative to audio features such as mel-filterbanks, that can be initialized as an approximation of mel-filterbanks…☆520Mar 1, 2022Updated 4 years ago