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 5 months ago
- A simple and intuitive library for diffusion models using Flax and Jax. Includes detailed notebooks on DDPM, DDIM, and EDM with simplifie…☆40Jul 27, 2026Updated 3 weeks ago
- A mini-library for training consistency models.☆251Dec 26, 2023Updated 2 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- EDM2 and Autoguidance -- Official PyTorch implementation☆850Dec 9, 2024Updated last year
- Official PyTorch implementation for Decomposed Diffusion Sampling (DDS), presented in the paper "Decomposed Diffusion Sampler for Acceler…☆82Sep 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,991Mar 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
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Pytorch implementation of "Light Schrödinger Bridge" (ICLR 2024)☆66Aug 9, 2024Updated 2 years ago
- ☆12Nov 7, 2024Updated last year
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆214Feb 19, 2025Updated last year
- ☆16Nov 20, 2023Updated 2 years ago
- [Science Advances] Data-driven Global Ocean Modeling for Seasonal to Decadal Prediction☆26Jul 23, 2026Updated last month
- [NeurIPS2021] Exploring Architectural Ingredients of Adversarially Robust Deep Neural Networks☆34Jul 5, 2024Updated 2 years ago
- Using generative AI for eddy-resolving multi-modal surface ocean state estimation☆19Jun 4, 2025Updated last year
- Short-time Fourier transform (STFT) for JAX☆15Dec 20, 2021Updated 4 years ago
- Information-Theoretic Diffusion: A brand new diffusion model / density estimator based on information theory.☆39Jul 18, 2024Updated 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.
- rabitq rust implementation☆11May 14, 2026Updated 3 months ago
- Score identity Distillation with Long and Short Guidance for One-Step Text-to-Image Generation☆96Dec 4, 2025Updated 8 months ago
- Official Pytorch implementation for "Tree-Based Diffusion Schrödinger Bridge with Applications to Wasserstein Barycenters" (NeurIPS 2023)☆10Mar 5, 2024Updated 2 years ago
- Pytorch wrappers for the FINUFFT library☆16Nov 21, 2025Updated 9 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
- [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"☆52Apr 26, 2024Updated 2 years ago
- Official PyTorch implementation for "Unifying Bayesian Flow Networks and Diffusion Models through Stochastic Differential Equations"☆41Apr 23, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- [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,129Jul 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☆61Nov 26, 2024Updated last year