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 2 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
- Neural Optimal Transport with Lagrangian Costs☆58Updated 4 months ago
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆24Updated last year
- ☆27Updated 2 years ago
- ☆38Updated 2 years ago
- ☆34Updated 5 months ago
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆61Updated 6 months ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated 2 years ago
- ☆31Updated last year
- Experiment with diffusion models that you can run on your local jupyter instances☆63Updated 11 months ago
- Variational Autoencoder with non-euclidean (hyperbolic) latent space☆11Updated 2 years ago
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- Meta Optimal Transport☆103Updated 2 years ago
- Transformers with doubly stochastic attention☆47Updated 3 years ago
- Simple Scalable Discrete Diffusion for text in PyTorch☆36Updated last year
- ☆31Updated last month
- [ICLR 2022] Path integral sampler☆52Updated 2 years ago
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆43Updated 3 months ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-1 transport computation (NeurIPS'22).☆21Updated last year
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆50Updated last year
- ☆44Updated last year
- Code for the paper: Rotating Features for Object Discovery☆53Updated last year
- Posterior Refinement Improves Sample Efficiency in Bayesian Neural Networks☆10Updated 2 years ago
- ☆17Updated 3 years ago
- The Superposition of Diffusion Models Using the Itô Density Estimator☆51Updated 6 months ago
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆30Updated last year
- NeurIPS23 "Flow Factorized Representation Learning"☆38Updated 11 months ago
- ☆17Updated 2 years ago
- ☆16Updated 2 years ago
- NeurIPS'23: Energy Discrepancies: A Score-Independent Loss for Energy-Based Models☆17Updated 11 months ago