fkodom / fft-conv-pytorchView external linksLinks
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.☆386May 25, 2024Updated last year
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆142Dec 21, 2021Updated 4 years ago
- DCT (discrete cosine transform) functions for pytorch☆640Nov 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 4 years ago
- Fast PyTorch based DSP for audio and 1D signals☆453Feb 17, 2025Updated 11 months ago
- Udacity CS344 Introduction to Parallell Programming (https://classroom.udacity.com/courses/cs344), with assignments/materials updated to …☆46Sep 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,157Aug 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).☆564Jul 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
- Implementation of the 😇 Attention layer from the paper, Scaling Local Self-Attention For Parameter Efficient Visual Backbones☆201Mar 24, 2021Updated 4 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,161Mar 22, 2024Updated last year
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,194Aug 22, 2023Updated 2 years ago
- 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'☆39Dec 4, 2021Updated 4 years ago
- Implementation of a Transformer, but completely in Triton☆279Apr 5, 2022Updated 3 years ago
- Source code for "On the Relationship between Self-Attention and Convolutional Layers"☆1,116Jan 10, 2023Updated 3 years ago
- Implementation of Long-Short Transformer, combining local and global inductive biases for attention over long sequences, in Pytorch☆120Aug 4, 2021Updated 4 years ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,395Jan 26, 2026Updated 2 weeks ago
- [NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classification☆500Jun 12, 2023Updated 2 years ago
- Official implementation of the NeuralSBS paper☆15Jun 17, 2021Updated 4 years ago
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022☆1,175May 15, 2024Updated last year
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆5,800Updated this week
- Implementation of Invariant Point Attention, used for coordinate refinement in the structure module of Alphafold2, as a standalone Pytorc…☆171Nov 25, 2022Updated 3 years ago
- Fully featured implementation of Routing Transformer☆300Nov 6, 2021Updated 4 years ago
- Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains☆1,362Jan 17, 2023Updated 3 years ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,473May 2, 2025Updated 9 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☆171May 15, 2020Updated 5 years ago
- Pytorch library for fast transformer implementations☆1,761Mar 23, 2023Updated 2 years ago
- Count the MACs / FLOPs of your PyTorch model.☆5,081Jul 8, 2024Updated last year
- High-fidelity performance metrics for generative models in PyTorch☆1,172Nov 18, 2025Updated 2 months ago
- ☆53May 15, 2025Updated 8 months ago