Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.
☆525Sep 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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- This is an official pytorch implementation of Fast Fourier Convolution.☆385May 25, 2024Updated 2 years ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆141Dec 21, 2021Updated 4 years ago
- Fast PyTorch based DSP for audio and 1D signals☆462Jun 3, 2026Updated 3 months ago
- DCT (discrete cosine transform) functions for pytorch☆644Nov 6, 2022Updated 3 years ago
- A temporal module for PyTorch-ComplexTensor☆44Jun 28, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Code for the paper "Unsupervised Contrastive Learning of Sound Event Representations", ICASSP 2021.☆93Dec 22, 2022Updated 3 years ago
- Udacity CS344 Introduction to Parallell Programming (https://classroom.udacity.com/courses/cs344), with assignments/materials updated to …☆47Sep 16, 2021Updated 5 years ago
- Pytorch implementation of 2D Discrete Wavelet (DWT) and Dual Tree Complex Wavelet Transforms (DTCWT) and a DTCWT based ScatterNet☆1,176Aug 2, 2023Updated 3 years ago
- Implementation of the 😇 Attention layer from the paper, Scaling Local Self-Attention For Parameter Efficient Visual Backbones☆199Mar 24, 2021Updated 5 years ago
- Implementation of a Transformer, but completely in Triton☆279Apr 5, 2022Updated 4 years ago
- Accompanying code for our paper "Optimizing Short-Time Fourier Transform Parameters via Gradient Descent"☆33Oct 30, 2020Updated 5 years ago
- Audio processing by using pytorch 1D convolution network☆1,127May 21, 2026Updated 3 months ago
- Relative Positional Encoding for Transformers with Linear Complexity☆65Apr 1, 2022Updated 4 years ago
- Asteroid's filterbanks☆90Jan 12, 2025Updated last year
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).☆566Jul 13, 2023Updated 3 years ago
- Implementation of the paper - Fast Training of Convolutional Networks through FFTs (CUDA for parallelization)☆10May 8, 2020Updated 6 years ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,604Aug 26, 2026Updated 3 weeks ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Dec 4, 2021Updated 4 years ago
- ☆53May 15, 2025Updated last year
- Fast, differentiable sorting and ranking in PyTorch☆865Mar 3, 2026Updated 6 months ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,170Mar 22, 2024Updated 2 years ago
- A pytorch implementation of FFTNet.☆37Aug 31, 2018Updated 8 years ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,485May 2, 2025Updated last year
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Official implementation of the NeuralSBS paper☆15Jun 17, 2021Updated 5 years ago
- Is the attention layer even necessary? (https://arxiv.org/abs/2105.02723)☆485May 7, 2021Updated 5 years ago
- Implementation of Long-Short Transformer, combining local and global inductive biases for attention over long sequences, in Pytorch☆120Aug 4, 2021Updated 5 years ago
- [NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classification☆511Jun 12, 2023Updated 3 years ago
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆5,949Updated this week
- Faster depthwise convolutions for PyTorch☆66Feb 20, 2020Updated 6 years ago
- A fast implementation of bss_eval metrics for blind source separation☆150Mar 11, 2026Updated 6 months ago
- Data-Driven Neuron Allocation for Scale Aggregation Networks☆54Apr 29, 2019Updated 7 years ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,217Updated this week
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains☆1,375Jan 17, 2023Updated 3 years ago
- ESPnet-TTS Audio Sample HP☆21Oct 25, 2019Updated 6 years ago
- Count the MACs / FLOPs of your PyTorch model.☆5,079Jul 8, 2024Updated 2 years ago
- Source code for "On the Relationship between Self-Attention and Convolutional Layers"☆1,121Jan 10, 2023Updated 3 years ago
- ☆10Sep 13, 2021Updated 5 years ago
- Implementation of Invariant Point Attention, used for coordinate refinement in the structure module of Alphafold2, as a standalone Pytorc…☆173Nov 25, 2022Updated 3 years ago
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022☆1,185May 15, 2024Updated 2 years ago