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.
☆25Updated last month
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:
- ☆34Updated 5 months ago
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆29Updated 6 months ago
- ☆26Updated 2 years ago
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆24Updated last year
- ☆38Updated 2 years ago
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆59Updated 6 months ago
- The Superposition of Diffusion Models Using the Itô Density Estimator☆52Updated 5 months ago
- ☆29Updated 3 weeks ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- Neural Optimal Transport with Lagrangian Costs☆57Updated 3 months ago
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆40Updated 3 months ago
- Transformers with doubly stochastic attention☆47Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated 2 years ago
- Source code accompanying the NeurIPS 2022 paper "Learning Partial Equivariances From Data"☆10Updated 2 years ago
- Code for the paper: Rotating Features for Object Discovery☆53Updated last year
- Perceiver (transformer variant) implemented in JAX and Flax☆12Updated 4 years ago
- ☆31Updated last year
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆51Updated 2 years ago
- [ICLR 2022] Path integral sampler☆50Updated 2 years ago
- ☆17Updated 3 years ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆65Updated 5 months ago
- Generative Equilibrium Transformer☆23Updated last year
- Variational Autoencoder with non-euclidean (hyperbolic) latent space☆11Updated 2 years ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆50Updated last year
- Posterior Refinement Improves Sample Efficiency in Bayesian Neural Networks☆10Updated 2 years ago
- ☆44Updated last year
- Simple Scalable Discrete Diffusion for text in PyTorch☆34Updated 11 months ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆45Updated 2 years ago
- Official implementation of Deep Momentum Schrödinger Bridge☆27Updated last year
- Meta Optimal Transport☆103Updated 2 years ago