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 last month
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
- ☆38Updated 2 years ago
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆24Updated last year
- ☆26Updated 2 years ago
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆58Updated 5 months ago
- Transformers with doubly stochastic attention☆46Updated 2 years ago
- Official implementation of Deep Momentum Schrödinger Bridge☆27Updated last year
- Euclidean Wasserstein-2 optimal transportation☆47Updated last year
- Variational Autoencoder with non-euclidean (hyperbolic) latent space☆11Updated 2 years ago
- ☆35Updated 4 months ago
- ☆29Updated this week
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆50Updated last year
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆29Updated 5 months ago
- Code for the paper: Rotating Features for Object Discovery☆53Updated last year
- ☆31Updated last year
- Posterior Refinement Improves Sample Efficiency in Bayesian Neural Networks☆10Updated 2 years ago
- NeurIPS23 "Flow Factorized Representation Learning"☆37Updated 9 months ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆24Updated last year
- Source code accompanying the NeurIPS 2022 paper "Learning Partial Equivariances From Data"☆10Updated 2 years ago
- Meta Optimal Transport☆103Updated 2 years ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-1 transport computation (NeurIPS'22).☆21Updated 11 months ago
- Neural Optimal Transport with Lagrangian Costs☆57Updated 2 months ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆45Updated 2 years ago
- The Superposition of Diffusion Models Using the Itô Density Estimator☆51Updated 4 months ago
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆30Updated last year
- ICLR25 | Official code base for Heavy-Tailed Diffusion with Denoising Levy Probabilistic Models (DLPM)☆17Updated 6 months ago
- Perceiver (transformer variant) implemented in JAX and Flax☆12Updated 4 years ago
- ☆18Updated last year
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆40Updated 2 months ago