Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist
☆67Dec 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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Minimal multi-gpu implementation of EDM2: "Analyzing and Improving the Training Dynamics of Diffusion Models"☆40Mar 5, 2024Updated 2 years ago
- 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☆36Mar 19, 2026Updated 2 months ago
- [Science Advances] Data-driven Global Ocean Modeling for Seasonal to Decadal Prediction☆23Sep 9, 2025Updated 8 months ago
- A mini-library for training consistency models.☆253Dec 26, 2023Updated 2 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- EDM2 and Autoguidance -- Official PyTorch implementation☆844Dec 9, 2024Updated last year
- Official PyTorch implementation for Decomposed Diffusion Sampling (DDS), presented in the paper "Decomposed Diffusion Sampler for Acceler…☆80Sep 26, 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 for Proximal Neural Networks.☆11Sep 17, 2021Updated 4 years ago
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,961Mar 16, 2024Updated 2 years ago
- ☆13Sep 13, 2023Updated 2 years ago
- Out-of-distribution generalization benchmarks for image recognition models☆14Apr 5, 2020Updated 6 years ago
- Diffusers reimplementation for https://rf-inversion.github.io/☆25Dec 17, 2024Updated last year
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆91Feb 14, 2025Updated last year
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- ☆44May 3, 2024Updated 2 years ago
- Pytorch implementation of "Light Schrödinger Bridge" (ICLR 2024)☆66Aug 9, 2024Updated last year
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆214Feb 19, 2025Updated last year
- ☆16Nov 20, 2023Updated 2 years ago
- Short-time Fourier transform (STFT) for JAX☆15Dec 20, 2021Updated 4 years ago
- rabitq rust implementation☆11May 14, 2026Updated 2 weeks ago
- Information-Theoretic Diffusion: A brand new diffusion model / density estimator based on information theory.☆38Jul 18, 2024Updated last year
- [ICCV 2025] Official Implementation of Contrastive Flow Matching☆179Jun 25, 2025Updated 11 months ago
- Tree-Based Diffusion Schrödinger Bridge with Applications to Wasserstein Barycenters☆10Mar 5, 2024Updated 2 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Consistency Models Made Easy☆330Oct 13, 2024Updated last year
- ☆53Jun 13, 2025Updated 11 months ago
- ☆20Nov 2, 2018Updated 7 years ago
- Official implementation of our paper "Bidirectional Consistency Models"; and reproduced Improved Consistency Models (iCT).☆27May 10, 2025Updated last year
- [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"☆53Apr 26, 2024Updated 2 years ago
- [ICLR 2024] Official implementation of Bellman Optimal Stepsize Straightening of Flow-Matching Models☆37Feb 25, 2024Updated 2 years ago
- Official PyTorch implementation for "Unifying Bayesian Flow Networks and Diffusion Models through Stochastic Differential Equations"☆42Apr 23, 2024Updated 2 years ago
- Maximize the Resolution Potential of Pre-trained Rectified Flow Transformers☆66Oct 16, 2024Updated last year
- PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)☆2,110Jul 14, 2024Updated 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.
- Differentiable and accelerated wavelet transform on the sphere with JAX☆26May 27, 2025Updated last year
- PyTorch implementation of One-step Diffusion with Distribution Matching Distillation☆65Jun 1, 2024Updated last year
- [ECCV2024] "SlimFlow: Training Smaller One-Step Diffusion Models with Rectified Flow", Yuanzhi Zhu, Xingchao Liu, Qiang Liu☆62Nov 26, 2024Updated last year
- A simple, easy-to-understand library for diffusion models using Flax and Jax. Includes detailed notebooks on DDPM, DDIM, and EDM with sim…☆41Apr 29, 2026Updated last month
- TiNO-Edit: Timestep and Noise Optimization for Robust Diffusion-Based Image Editing (CVPR 2024)☆44Sep 12, 2025Updated 8 months ago
- [ICCV 2023] Efficient Diffusion Training via Min-SNR Weighting Strategy☆269Dec 10, 2024Updated last year
- ☆33Feb 14, 2024Updated 2 years ago