v0lta / PyTorch-Wavelet-Toolbox
Differentiable fast wavelet transforms in PyTorch with GPU support.
☆352Updated last month
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,064Updated last year
- Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.☆494Updated last year
- PyTorch library for solving imaging inverse problems using deep learning☆429Updated this week
- This is an official pytorch implementation of Fast Fourier Convolution.☆371Updated 11 months ago
- Code repository of the paper "Modelling Long Range Dependencies in ND: From Task-Specific to a General Purpose CNN" https://arxiv.org/abs…☆184Updated last week
- Multi-level Wavelet Convolutional Neural Networks☆287Updated 3 years ago
- Official implementation of "DiffuseVAE: Efficient, Controllable and High-Fidelity Generation from Low-Dimensional Latents"☆367Updated 2 years ago
- Official pytorch repository for "Diffusion Posterior Sampling for General Noisy Inverse Problems"☆501Updated 2 years ago
- Adaptive wavelet pooling for CNN in PyTorch, AISTATS 2021.☆33Updated 3 years ago
- Official Pytorch Implementation of the paper: Wavelet Diffusion Models are fast and scalable Image Generators (CVPR'23)☆408Updated 9 months ago
- Convolutional layer for Kolmogorov-Arnold Network (KAN)☆99Updated last month
- This repository implements time series diffusion in the frequency domain.☆45Updated 2 months ago
- [NeurIPS 2022] Denoising Diffusion Restoration Models -- Official Code Repository☆632Updated 2 years ago
- The official PyTorch implementation for NCSNv2 (NeurIPS 2020)☆304Updated 3 years ago
- Fast implementation of the multi-dimensional fractional Fourier transform in Python☆49Updated 2 years ago
- Official PyTorch implementation of the NeurIPS 2022 paper "Improving Diffusion Models for Inverse Problems using Manifold Constraints (ht…☆243Updated 2 years ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆582Updated 5 months ago
- Official PyTorch implementation for the paper "CARD: Classification and Regression Diffusion Models"☆227Updated 2 years ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆279Updated last year
- DCT (discrete cosine transform) functions for pytorch☆610Updated 2 years ago
- PyTorch implementation of normalizing flow models☆834Updated 8 months ago
- PyTorch implementation of the wavelet analysis from Torrence & Compo (1998)☆315Updated 3 years ago
- Adaptive, interpretable wavelets across domains (NeurIPS 2021)☆77Updated 3 years ago
- Normalizing flows in PyTorch☆380Updated last month
- Official repo for "Solving Inverse Problems in Medical Imaging with Score-Based Generative Models"☆245Updated 2 years ago
- PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.☆65Updated 2 years ago
- Fast Convolutional KAN☆58Updated last year
- Official implementation of Diffusion Autoencoders☆918Updated 8 months ago
- This repository contains the codes to replicate the simulations from the paper: "Wav-KAN: Wavelet Kolmogorov-Arnold Networks". It showca…☆142Updated 2 months ago
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆154Updated 2 months ago