yuanzhi-zhu / mini_edmView external linksLinks
Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist
☆64Dec 16, 2023Updated 2 years ago
Alternatives and similar repositories for mini_edm
Users that are interested in mini_edm are comparing it to the libraries listed below
Sorting:
- Minimal multi-gpu implementation of EDM2: "Analyzing and Improving the Training Dynamics of Diffusion Models"☆40Mar 5, 2024Updated last year
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆97Feb 12, 2024Updated 2 years ago
- Official implementation of Learning Diffusion Priors from Observations by Expectation Maximization☆34Oct 31, 2025Updated 3 months ago
- ☆13Sep 13, 2023Updated 2 years ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆819Dec 9, 2024Updated last year
- Official PyTorch implementation for Decomposed Diffusion Sampling (DDS), presented in the paper "Decomposed Diffusion Sampler for Acceler…☆75Sep 26, 2024Updated last year
- A mini-library for training consistency models.☆251Dec 26, 2023Updated 2 years ago
- [ICLR 2024] Official implementation of Bellman Optimal Stepsize Straightening of Flow-Matching Models☆37Feb 25, 2024Updated last year
- Information-Theoretic Diffusion: A brand new diffusion model / density estimator based on information theory.☆37Jul 18, 2024Updated last year
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆201Feb 19, 2025Updated 11 months ago
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆91Feb 14, 2025Updated last year
- Pytorch implementation of "Light Schrödinger Bridge" (ICLR 2024)☆64Aug 9, 2024Updated last year
- ☆39Apr 27, 2024Updated last year
- ☆44May 3, 2024Updated last year
- PyTorch-based radio-interferometric imaging reconstruction package with scalable Bayesian uncertainty quantification relying on data-driv…☆12Feb 17, 2025Updated last year
- Implementation of Diffusion Transformers and Rectified Flow in Jax☆27Jul 9, 2024Updated last year
- Implementation for Proximal Neural Networks.☆11Sep 17, 2021Updated 4 years ago
- ☆11Nov 7, 2024Updated last year
- [ICCV 2025] Official Implementation of Contrastive Flow Matching☆160Jun 25, 2025Updated 7 months ago
- ☆15Nov 20, 2023Updated 2 years ago
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,905Mar 16, 2024Updated last year
- Code for our papers : "Generating images of rare concepts using pre-trained diffusion models" (AAAI 24) and "Norm-guided latent space exp…☆86Dec 27, 2023Updated 2 years ago
- Short-time Fourier transform (STFT) for JAX☆15Dec 20, 2021Updated 4 years ago
- [ECCV2024] "SlimFlow: Training Smaller One-Step Diffusion Models with Rectified Flow", Yuanzhi Zhu, Xingchao Liu, Qiang Liu☆60Nov 26, 2024Updated last year
- ☆13Nov 13, 2020Updated 5 years ago
- DiffPhase: Generative Diffusion-based STFT Phase Retrieval☆16Sep 21, 2023Updated 2 years ago
- ☆17Aug 13, 2024Updated last year
- Github repository for the paper accepted in ICASSP 2024 : Blind estimation of audio effects using an auto-encoder approach and differenti…☆14Apr 11, 2024Updated last year
- ☆14Mar 16, 2023Updated 2 years ago
- Consistency Models Made Easy☆325Oct 13, 2024Updated last year
- ESLTTS dataset☆16Feb 6, 2025Updated last year
- Official implementation of the paper "Normalization-Equivariant Neural Networks with Application to Image Denoising" (NeurIPS'23)☆13Jun 27, 2025Updated 7 months ago
- PyTorch implementation of simplified neural source filter model (s-nsf)☆14Aug 4, 2021Updated 4 years ago
- [ICCV 2023] Efficient Diffusion Training via Min-SNR Weighting Strategy☆265Dec 10, 2024Updated last year
- DLD: A Deep Learning Based Line Descriptor for Line Feature Matching☆18Nov 21, 2022Updated 3 years ago
- Visualising Losses in Deep Neural Networks☆16Jul 17, 2024Updated last year
- Pytorch wrappers for the FINUFFT library☆16Nov 21, 2025Updated 2 months ago
- Flow-matching algorithms in JAX☆115Aug 12, 2024Updated last year
- Vocal Remover using Deep Neural Networks☆19Dec 31, 2024Updated last year