A minimal, single-file implementation of the MeanFlow paper on 2D toy examples, with a side-by-side comparison to rectified flow.
☆26Jul 4, 2025Updated 8 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:
- A Python library for variational inference with normalizing flow and annealing☆15Mar 13, 2025Updated 11 months ago
- Minimal JAX implementation unifying Diffusion and Flow Matching algorithms as alternative strategies for transporting data distributions.☆63Dec 19, 2025Updated 2 months ago
- ☆15Mar 30, 2020Updated 5 years ago
- ☆18Jun 26, 2023Updated 2 years ago
- Discovering Quality-Diversity Algorithms via Meta-Black-Box Optimization☆21Dec 1, 2025Updated 3 months ago
- Euclidean Wasserstein-2 optimal transportation☆46Aug 19, 2023Updated 2 years ago
- ☆18Mar 16, 2022Updated 3 years ago
- This is code to accompany the paper "Accelerating Exploration with Unlabeled Prior Data".☆25Dec 5, 2023Updated 2 years ago
- Definitive implementation of the stochastic interpolant framework for generative modeling in jax.☆40Jul 13, 2025Updated 7 months ago
- Distributional Sliced-Wasserstein distance code☆50Jul 22, 2024Updated last year
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆31Mar 21, 2024Updated last year
- Cross-Domain Imitation Learning via Optimal Transport☆25Jun 24, 2022Updated 3 years ago
- Hyperbolic Busemann Learning with Ideal Prototypes, NeurIPS2021☆24Dec 9, 2021Updated 4 years ago
- Embedding graphs in symmetric spaces☆30Sep 30, 2021Updated 4 years ago
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆71Mar 4, 2025Updated last year
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆37Mar 18, 2025Updated 11 months ago
- Repository for the paper "Optimal design of stochastic DNA synthesis protocols based on generative sequence models" (Weinstein et al., AI…☆36Feb 17, 2022Updated 4 years ago
- rebuilds and completes models of protein complexes using AlphaFold2☆16Feb 26, 2026Updated last week
- Dataset corresponding to the paper: "Form2Seq : A Framework for Higher-Order Form Structure Extraction"☆10Feb 17, 2021Updated 5 years ago
- Source code for the ICML2019 paper "Subspace Robust Wasserstein Distances"☆29Jun 14, 2019Updated 6 years ago
- ☆33May 31, 2019Updated 6 years ago
- ☆10Apr 30, 2025Updated 10 months ago
- Husky-LIO-SAM☆11Feb 23, 2023Updated 3 years ago
- Protein scoring and sampling of 'Combinatorial Variant Effects from Structure' (CoVES)☆11Jan 5, 2024Updated 2 years ago
- Source code accompanying the NeurIPS 2022 paper "Learning Partial Equivariances From Data"☆10Nov 18, 2022Updated 3 years ago
- A Structural Average of Labeled Merge Trees for Uncertainty Visualization☆11Apr 22, 2024Updated last year
- A nonparametric variational information bottleneck (NVIB) layer in Pytorch☆11Apr 15, 2025Updated 10 months ago
- ☆10Dec 8, 2023Updated 2 years ago
- hierarchical core-periphery structure☆10Jul 21, 2023Updated 2 years ago
- ☆12Apr 9, 2025Updated 11 months ago
- Demonstration of the UPGMA hierarchal clustering algorithm in Pandas, Seaborn, and Scipy☆11Sep 29, 2019Updated 6 years ago
- [IROS 2025] EgoLoc: Zero-Shot Temporal Interaction Localization for Egocentric Videos☆33Jan 13, 2026Updated last month
- Implementations of the renormalization group-based diffusion model (RGDM).☆16Mar 10, 2025Updated 11 months ago
- [ICML2023] InfoOT: Information Maximizing Optimal Transport☆41Apr 27, 2023Updated 2 years ago
- ☆13Jun 5, 2024Updated last year
- ☆10Jun 14, 2023Updated 2 years ago
- ☆10Jun 24, 2023Updated 2 years ago
- Getting Started in Imitation Learning☆13Mar 3, 2025Updated last year
- ☆12Mar 15, 2023Updated 2 years ago