Alternatives and similar repositories for Large-Scale-Wasserstein-Gradient-Flows

Users that are interested in Large-Scale-Wasserstein-Gradient-Flows are comparing it to the libraries listed below

• milenagazdieva / LightUnbalancedOptimalTransport

  View on GitHub
  PyTorch implementation of "Light Unbalanced Optimal Transport" (NeurIPS 2024)
  ☆22Dec 23, 2024Updated last year
• milenagazdieva / U-NOTBarycenters

  View on GitHub
  PyTorch implementation of "Robust Barycenter Estimation using Semi-unbalanced Neural Optimal Transport" (ICLR 2025)
  ☆17Jul 7, 2025Updated 7 months ago
• sbyebss / variational_wgf

  View on GitHub
  [ICML2022] Variational Wasserstein gradient flow
  ☆24Oct 17, 2022Updated 3 years ago
• Daniil-Selikhanovych / ASBM

  View on GitHub
  PyTorch implementation of "Adversarial Schrödinger Bridge Matching" (NeurIPS 2024)
  ☆28Jun 2, 2025Updated 8 months ago
• bunnech / jkonet

  View on GitHub
  Proximal Optimal Transport Modeling of Population Dynamics (AISTATS 2022)
  ☆22Jun 19, 2023Updated 2 years ago
• ngushchin / LightSB

  View on GitHub
  Pytorch implementation of "Light Schrödinger Bridge" (ICLR 2024)
  ☆64Aug 9, 2024Updated last year
• iamalexkorotin / KernelNeuralOptimalTransport

  View on GitHub
  PyTorch implementation of "Kernel Neural Optimal Transport" (ICLR 2023)
  ☆29Oct 3, 2023Updated 2 years ago
• iamalexkorotin / Wasserstein2Barycenters

  View on GitHub
  PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)
  ☆33Jun 17, 2022Updated 3 years ago
• iamalexkorotin / Wasserstein2Benchmark

  View on GitHub
  A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation (NeurIPS 2021)
  ☆43Jun 17, 2022Updated 3 years ago
• milenagazdieva / ExtremalNeuralOptimalTransport

  View on GitHub
  PyTorch implementation of "Extremal Domain Translation with Neural Optimal Transport" (NeurIPS 2023)
  ☆21Dec 24, 2024Updated last year
• SKholkin / LightSB-Matching

  View on GitHub
  Light and Optimal Schrödinger Bridge Matching (ICML 2024) official PyTorch implementation
  ☆57Aug 8, 2024Updated last year
• necludov / oMCMC

  View on GitHub
  orbital MCMC
  ☆10Jun 17, 2021Updated 4 years ago
• FabianAltekrueger / NeuralWassersteinGradientFlows

  View on GitHub
  Implementation of the paper "Neural Wasserstein Gradient Flows for Discrepancies with Riesz Kernels"
  ☆11Mar 18, 2024Updated last year
• CW-Huang / CP-Flow

  View on GitHub
  Convex potential flows
  ☆85Nov 19, 2021Updated 4 years ago
• bunnech / condot

  View on GitHub
  Supervised Training of Conditional Monge Maps
  ☆19Oct 30, 2023Updated 2 years ago
• antonioterpin / jkonet-star

  View on GitHub
  JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".
  ☆37Mar 18, 2025Updated 10 months ago
• ngushchin / EntropicNeuralOptimalTransport

  View on GitHub
  Pytorch implementation of "Entropic Neural Optimal Transport via Diffusion Processes" (NeurIPS 2023, oral).
  ☆42Mar 11, 2024Updated last year
• ngushchin / EntropicOTBenchmark

  View on GitHub
  Entropic Optimal Transport Benchmark (NeurIPS 2023).
  ☆27Mar 21, 2024Updated last year
• pnkraemer / million-dimension-prob-ode-solver-experiments

  View on GitHub
  ☆10Jun 13, 2022Updated 3 years ago
• facebookresearch / w2ot

  View on GitHub
  Euclidean Wasserstein-2 optimal transportation
  ☆46Aug 19, 2023Updated 2 years ago
• JTT94 / diffusion_schrodinger_bridge

  View on GitHub
  PyTorch Implementation of DSB for Score Based Generative Modeling. Experiments managed using Hydra.
  ☆191Nov 23, 2021Updated 4 years ago
• iamalexkorotin / NeuralOptimalTransport

  View on GitHub
  PyTorch implementation of "Neural Optimal Transport" (ICLR 2023 Spotlight)
  ☆226Dec 11, 2023Updated 2 years ago
• lingxiaoli94 / CWB

  View on GitHub
  Source code for "Continuous Regularized Wasserstein Barycenters" [NeurIPS 2020].
  ☆15Nov 4, 2020Updated 5 years ago
• jshe / wasserstein-2

  View on GitHub
  ☆13Mar 26, 2019Updated 6 years ago
• samuelepapa / neural-field-arena

  View on GitHub
  ☆12Jan 9, 2024Updated 2 years ago
• clementchadebec / geometric_perspective_on_vaes

  View on GitHub
  Code of "A Geometric Perspective on Variational Autoencoders" (NeurIPS 2022)
  ☆14Nov 19, 2024Updated last year
• samcohen16 / Aligning-Time-Series

  View on GitHub
  ☆51Jul 25, 2024Updated last year
• DoronHav / WassersteinFlowMatching

  View on GitHub
  ☆39Jan 27, 2026Updated 2 weeks ago
• desResLab / LINFA

  View on GitHub
  A Python library for variational inference with normalizing flow and annealing
  ☆15Mar 13, 2025Updated 11 months ago
• AndyShih12 / mac

  View on GitHub
  PyTorch implementation for "Training and Inference on Any-Order Autoregressive Models the Right Way", NeurIPS 2022 Oral, TPM 2023 Best Pa…
  ☆15May 31, 2023Updated 2 years ago
• mshakerinava / AutoEquiv

  View on GitHub
  Group Equivariant Linear Maps
  ☆11Jun 4, 2021Updated 4 years ago
• PhIMaL / temporal_normalizing_flows

  View on GitHub
  ☆11Apr 2, 2021Updated 4 years ago
• NickHunt-Smith / MCMC-diffusion

  View on GitHub
  A Metropolis-Hastings MCMC sampler accelerated via diffusion models
  ☆16Jul 25, 2024Updated last year
• EmoryMLIP / OT-Flow

  View on GitHub
  PyTorch implementation of the OT-Flow approach in arXiv:2006.00104
  ☆57Jul 25, 2024Updated last year
• ExplainableML / uot-fm

  View on GitHub
  Official Repository for "Unbalancedness in Neural Monge Maps Improves Unpaired Domain Translation" [ICLR 2024]
  ☆17May 15, 2024Updated last year
• necludov / wqmc

  View on GitHub
  ☆14Jul 17, 2023Updated 2 years ago
• tongzhoumu / mean_flow_2d

  View on GitHub
  A minimal, single-file implementation of the MeanFlow paper on 2D toy examples, with a side-by-side comparison to rectified flow.
  ☆27Jul 4, 2025Updated 7 months ago
• AkinoriTanaka-phys / DOT

  View on GitHub
  code submission to NeurIPS2019
  ☆13Aug 9, 2023Updated 2 years ago
• ChengzijunAixiaoli / PPMM

  View on GitHub
  Python3 implementation of the paper [Large-scale optimal transport map estimation using projection pursuit]
  ☆15Feb 24, 2021Updated 4 years ago