myscience / modular-diffusionLinks
Modular PyTorch (Lightning) implementation of Diffusion Probabilistic Models
☆21Updated 2 years ago
Alternatives and similar repositories for modular-diffusion
Users that are interested in modular-diffusion are comparing it to the libraries listed below
Sorting:
- ☆146Updated last year
- Flow-matching algorithms in JAX☆92Updated 9 months ago
- Code and experiments for the NeurIPS 2023 paper Stabilized Neural Differential Equations for Learning Dynamics with Explicit Constraints☆11Updated last year
- Code release for the paper Generative Modeling With Inverse Heat Dissipation☆70Updated 2 years ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- Official code for Continuous-Time Functional Diffusion Processes (NeurIPS 2023).☆19Updated last year
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆158Updated 3 months ago
- PyTorch implementation of Variational Diffusion Models.☆82Updated last year
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆69Updated last year
- ☆188Updated 5 months ago
- ☆28Updated last week
- Code for PnP-Flow: Plug-and-Play image restoration with Flow Matching (ICLR 2025)☆82Updated 3 months ago
- Educational implementation of the Discrete Flow Matching paper☆88Updated 9 months ago
- Score-Based Generative Modeling with Critically-Damped Langevin Diffusion☆197Updated last year
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆294Updated last week
- 🚀 A powerful library for efficient training of Neural Fields at scale.☆29Updated last year
- ☆13Updated 4 months ago
- Collecting research materials on neural samplers with diffusion/flow models☆48Updated this week
- code for "Riemannian Flow Matching on General Geometries".☆227Updated last year
- diffusionjax is a simple and accessible diffusion models package in JAX☆45Updated 4 months ago
- Implementation of papers in 101 lines of code.☆18Updated last year
- Unofficial minimal implementation of consistency models (CM) proposed by Song et al. 2023 on a 1D toy task in pytorch☆21Updated 2 years ago
- Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist☆53Updated last year
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆94Updated last year
- PyTorch Implementation of DSB for Score Based Generative Modeling. Experiments managed using Hydra.☆168Updated 3 years ago
- Score-based generative models for compact manifolds☆110Updated last year
- Diffusion models in PyTorch☆102Updated this week
- Official PyTorch implementation of the paper: Flow Matching in Latent Space☆271Updated 4 months ago
- Matrix-free linear algebra in JAX.☆119Updated last week
- InverseBench (ICLR 2025 spotlight)☆49Updated last month