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.
☆21Updated 2 weeks 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:
- Source code accompanying the NeurIPS 2022 paper "Learning Partial Equivariances From Data"☆10Updated 2 years ago
- ☆35Updated 3 months 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☆56Updated 2 months ago
- ☆29Updated last month
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆23Updated last year
- Code for the paper: Rotating Features for Object Discovery☆52Updated 11 months ago
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆39Updated last month
- ☆24Updated last year
- ☆44Updated last year
- ☆38Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated last year
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆45Updated 2 years ago
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆30Updated last year
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆56Updated 4 months ago
- [ICLR 2022] Path integral sampler☆48Updated last year
- The Superposition of Diffusion Models Using the Itô Density Estimator☆45Updated 4 months ago
- Meta Optimal Transport☆103Updated 2 years ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆24Updated last year
- NeurIPS23 "Flow Factorized Representation Learning"☆37Updated 8 months ago
- ☆33Updated 2 years ago
- NF-Layers for constructing neural functionals.☆87Updated last year
- ☆16Updated 2 years ago
- ICLR22 "Fast Differentiable Matrix Square Root" and T-PAMI extension☆64Updated 8 months ago
- Mirror Diffusion Models, NeurIPS 2023☆31Updated last year
- Transformers with doubly stochastic attention☆46Updated 2 years ago
- Official PyTorch and JAX Implementation of "Harmonics of Learning: Universal Fourier Features Emerge in Invariant Networks"☆38Updated last year
- Meta in-context learning for protein fitness prediction☆11Updated 5 months ago
- ☆17Updated 2 years ago