tongzhoumu / mean_flow_2dLinks
A minimal, single-file implementation of the MeanFlow paper on 2D toy examples, with a side-by-side comparison to rectified flow.
☆27Updated 6 months ago
Alternatives and similar repositories for mean_flow_2d
Users that are interested in mean_flow_2d are comparing it to the libraries listed below
Sorting:
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- ☆28Updated 2 years ago
- ☆36Updated 9 months ago
- ☆36Updated this week
- Transformers with doubly stochastic attention☆52Updated 3 years ago
- ☆39Updated 3 years ago
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆25Updated 2 years ago
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆30Updated 10 months ago
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆69Updated 10 months ago
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆52Updated 3 years ago
- ☆31Updated last year
- Implementations of the renormalization group-based diffusion model (RGDM).☆15Updated 10 months ago
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆30Updated last year
- Deep Networks Grok All the Time and Here is Why☆38Updated last year
- [ICLR 2022] Path integral sampler☆52Updated 2 years ago
- Neural Optimal Transport with Lagrangian Costs☆61Updated 8 months ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆51Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆66Updated last year
- ☆18Updated 2 years ago
- The Superposition of Diffusion Models Using the Itô Density Estimator☆51Updated 10 months ago
- Official implementation of Deep Momentum Schrödinger Bridge☆28Updated 2 years ago
- ☆18Updated last year
- ☆21Updated 11 months ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆71Updated 10 months ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated 2 years ago
- Code release for "Stochastic Optimal Control Matching"☆39Updated last year
- ☆18Updated 3 years ago
- Source code accompanying the NeurIPS 2022 paper "Learning Partial Equivariances From Data"☆10Updated 3 years ago
- Code for the paper: Rotating Features for Object Discovery☆53Updated last year
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆47Updated 2 years ago