Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist
☆66Dec 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"☆41Mar 5, 2024Updated 2 years ago
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆96Feb 12, 2024Updated 2 years ago
- Official implementation of Learning Diffusion Priors from Observations by Expectation Maximization☆37Mar 19, 2026Updated 4 months ago
- A simple and intuitive library for diffusion models using Flax and Jax. Includes detailed notebooks on DDPM, DDIM, and EDM with simplifie…☆41Updated this week
- A mini-library for training consistency models.☆251Dec 26, 2023Updated 2 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- EDM2 and Autoguidance -- Official PyTorch implementation☆848Dec 9, 2024Updated last year
- Official PyTorch implementation for Decomposed Diffusion Sampling (DDS), presented in the paper "Decomposed Diffusion Sampler for Acceler…☆81Sep 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
- ☆13Sep 13, 2023Updated 2 years ago
- ☆40Apr 27, 2024Updated 2 years ago
- Elucidating the Design Space of Diffusion-Based Generative Models (EDM)☆1,982Mar 16, 2024Updated 2 years ago
- Out-of-distribution generalization benchmarks for image recognition models☆14Apr 5, 2020Updated 6 years ago
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆91Feb 14, 2025Updated 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.
- ☆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☆216Feb 19, 2025Updated last year
- Official implementation of "Wave-Trainer-Fit: Neural Vocoder with Trainable Prior and Fixed-Point Iteration towards High-Quality Speech G…☆16Feb 6, 2026Updated 5 months ago
- [CVPR2025] MetricGrids: Arbitrary Nonlinear Approximation with Elementary Metric Grids based Implicit Neural Representation☆16Jun 1, 2025Updated last year
- ☆16Nov 20, 2023Updated 2 years ago
- [NeurIPS2021] Exploring Architectural Ingredients of Adversarially Robust Deep Neural Networks☆33Jul 5, 2024Updated 2 years ago
- Using generative AI for eddy-resolving multi-modal surface ocean state estimation☆17Jun 4, 2025Updated last year
- Short-time Fourier transform (STFT) for JAX☆15Dec 20, 2021Updated 4 years ago
- 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.
- Information-Theoretic Diffusion: A brand new diffusion model / density estimator based on information theory.☆38Jul 18, 2024Updated 2 years ago
- rabitq rust implementation☆11May 14, 2026Updated 2 months ago
- Score identity Distillation with Long and Short Guidance for One-Step Text-to-Image Generation☆96Dec 4, 2025Updated 7 months ago
- Tree-Based Diffusion Schrödinger Bridge with Applications to Wasserstein Barycenters☆10Mar 5, 2024Updated 2 years ago
- Pytorch wrappers for the FINUFFT library☆16Nov 21, 2025Updated 8 months ago
- ☆20Nov 2, 2018Updated 7 years ago
- Consistency Models Made Easy☆333Oct 13, 2024Updated last year
- Official implementation of our paper "Bidirectional Consistency Models"; and reproduced Improved Consistency Models (iCT).☆27May 10, 2025Updated last year
- Official PyTorch implementation for "Unifying Bayesian Flow Networks and Diffusion Models through Stochastic Differential Equations"☆41Apr 23, 2024Updated 2 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- [ICLR 2024] Official implementation of Bellman Optimal Stepsize Straightening of Flow-Matching Models☆37Feb 25, 2024Updated 2 years ago
- ESLTTS dataset☆16Feb 6, 2025Updated last year
- ☆19Aug 30, 2023Updated 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,124Jul 14, 2024Updated 2 years ago
- A demo shows how to combine Langevin dynamics with score matching for generative models.☆41Nov 19, 2020Updated 5 years ago
- [ECCV2024] "SlimFlow: Training Smaller One-Step Diffusion Models with Rectified Flow", Yuanzhi Zhu, Xingchao Liu, Qiang Liu☆62Nov 26, 2024Updated last year