v0lta / PyTorch-Wavelet-ToolboxLinks
Differentiable fast wavelet transforms in PyTorch with GPU support.
☆382Updated 3 weeks ago
Alternatives and similar repositories for PyTorch-Wavelet-Toolbox
Users that are interested in PyTorch-Wavelet-Toolbox are comparing it to the libraries listed below
Sorting:
- Pytorch implementation of 2D Discrete Wavelet (DWT) and Dual Tree Complex Wavelet Transforms (DTCWT) and a DTCWT based ScatterNet☆1,127Updated 2 years ago
- Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.☆508Updated 2 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…☆183Updated 4 months ago
- This is an official pytorch implementation of Fast Fourier Convolution.☆377Updated last year
- Fast implementation of the multi-dimensional fractional Fourier transform in Python☆51Updated 3 years ago
- Convolutional layer for Kolmogorov-Arnold Network (KAN)☆113Updated 6 months ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆288Updated last year
- DCT (discrete cosine transform) functions for pytorch☆629Updated 2 years ago
- Implementation of Depthwise Separable Convolution (pytorch)☆81Updated 5 years ago
- Official Pytorch Implementation of the paper: Wavelet Diffusion Models are fast and scalable Image Generators (CVPR'23)☆423Updated last year
- Multi-level Wavelet Convolutional Neural Networks☆296Updated 4 years ago
- Fast Convolutional KAN☆64Updated last year
- This repository contains the codes to replicate the simulations from the paper: "Wav-KAN: Wavelet Kolmogorov-Arnold Networks". It showca…☆151Updated 3 months ago
- ☆57Updated 3 months ago
- 1D interpolation for pytorch☆173Updated last year
- PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.☆72Updated 2 years ago
- A python/pytorch package for invertible neural networks☆67Updated 2 years ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆611Updated 10 months ago
- [NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classification☆487Updated 2 years ago
- It contains the code for the paper Deep Adaptive Wavelet Network☆73Updated 3 years ago
- Adaptive, interpretable wavelets across domains (NeurIPS 2021)☆81Updated 3 years ago
- DeepInverse: a PyTorch library for solving imaging inverse problems using deep learning☆523Updated this week
- Unofficial Implementation of MLP-Mixer, gMLP, resMLP, Vision Permutator, S2MLP, S2MLPv2, RaftMLP, HireMLP, ConvMLP, AS-MLP, SparseMLP, Co…☆169Updated 3 years ago
- Implementation of "Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains" by Tancik et al.☆100Updated last week
- A simple PyTorch implementation of conditional denoising diffusion probabilistic models (DDPM) on MNIST, Fashion-MNIST, and Sprite datase…☆75Updated last year
- Official repo for "Solving Inverse Problems in Medical Imaging with Score-Based Generative Models"☆255Updated 3 years ago
- KAE : KAN-based AutoEncoder (AE, VAE, VQ-VAE, RVQ, etc.)☆36Updated last year
- Description： Frequency Augmented Variational Autoencoder for better Image Reconstruction☆41Updated last year
- ☆136Updated last year
- Benchmark for efficiency in memory and time of different KAN implementations.☆133Updated last year