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 3 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
- ☆34Updated 2 months ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated 2 years ago
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆25Updated 2 years ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆30Updated last year
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆29Updated 8 months ago
- ☆33Updated 7 months ago
- ☆17Updated 3 years ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆24Updated last year
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆66Updated 7 months ago
- Official implementation of Deep Momentum Schrödinger Bridge☆28Updated last year
- Neural Optimal Transport with Lagrangian Costs☆59Updated 5 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆65Updated last year
- ☆18Updated last year
- Meta Optimal Transport☆104Updated 2 years ago
- Transformers with doubly stochastic attention☆49Updated 3 years ago
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆27Updated last year
- Code for the paper: Rotating Features for Object Discovery☆53Updated last year
- ☆31Updated last year
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆52Updated 3 years ago
- Posterior Refinement Improves Sample Efficiency in Bayesian Neural Networks☆10Updated 3 years ago
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆104Updated 2 weeks ago
- Implementation of Particle Guidance: non-I.I.D. Diverse Sampling with Diffusion Models☆71Updated 2 years ago
- Perceiver (transformer variant) implemented in JAX and Flax☆12Updated 4 years ago
- ☆39Updated 2 years ago
- Source code accompanying the NeurIPS 2022 paper "Learning Partial Equivariances From Data"☆10Updated 2 years ago
- Implementation of Action Matching☆46Updated 2 years ago
- NF-Layers for constructing neural functionals.☆90Updated last year
- ☆20Updated 8 months ago