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 5 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:
- ☆27Updated 2 years ago
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆69Updated 9 months ago
- ☆35Updated last week
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆29Updated 9 months ago
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆25Updated 2 years ago
- ☆35Updated 8 months ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated 2 years ago
- ☆31Updated last year
- Neural Optimal Transport with Lagrangian Costs☆60Updated 6 months ago
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆27Updated last year
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆52Updated 3 years ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆26Updated last year
- Official implementation of Metric Flow Matching (NeurIPS 2024)☆57Updated 3 months ago
- ☆44Updated last year
- ☆39Updated 3 years ago
- Official implementation of Deep Momentum Schrödinger Bridge☆27Updated 2 years ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆51Updated last year
- Code release for "Stochastic Optimal Control Matching"☆39Updated last year
- ☆18Updated 2 years ago
- Transformers with doubly stochastic attention☆50Updated 3 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆66Updated last year
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆30Updated last year
- The Superposition of Diffusion Models Using the Itô Density Estimator☆51Updated 8 months ago
- PyTorch implementation of "Kernel Neural Optimal Transport" (ICLR 2023)☆28Updated 2 years ago
- Code for the paper: Rotating Features for Object Discovery☆53Updated last year
- Source code accompanying the NeurIPS 2022 paper "Learning Partial Equivariances From Data"☆10Updated 3 years ago
- [ICLR 2022] Path integral sampler☆52Updated 2 years ago
- ☆18Updated last year
- Implementation of Action Matching for the Schrödinger equation☆25Updated 2 years ago