fkodom / fft-conv-pytorchLinks
Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.
☆497Updated last year
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:
- Implementation of ConvMixer for "Patches Are All You Need? 🤷"☆1,072Updated 2 years ago
- An All-MLP solution for Vision, from Google AI☆1,022Updated 8 months ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆590Updated 6 months ago
- DCT (discrete cosine transform) functions for pytorch☆612Updated 2 years ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.