Alternatives and similar repositories for jkonet

Users that are interested in jkonet are comparing it to the libraries listed below

• PetrMokrov / Large-Scale-Wasserstein-Gradient-Flows

  View on GitHub
  Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)
  ☆42May 8, 2022Updated 3 years ago
• take-koshizuka / NLSB

  View on GitHub
  ☆22Feb 4, 2025Updated last year
• mingyan08 / PD3O

  View on GitHub
  This is the class in matlab for convex optimization algorithms
  ☆10Nov 19, 2023Updated 2 years ago
• KrishnaswamyLab / Cell-Dynamics-Pipeline

  View on GitHub
  ☆10Dec 8, 2023Updated 2 years ago
• atong01 / MultiscaleEMD

  View on GitHub
  Multiscale earth mover's distance methods for tree and manifold structured data
  ☆12Mar 8, 2025Updated 11 months ago
• necludov / action-matching

  View on GitHub
  Implementation of Action Matching for the Schrödinger equation
  ☆25Jun 18, 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
• lingxiaoli94 / CWB

  View on GitHub
  Source code for "Continuous Regularized Wasserstein Barycenters" [NeurIPS 2020].
  ☆15Nov 4, 2020Updated 5 years ago
• 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
• KrishnaswamyLab / DiffusionEMD

  View on GitHub
  Code corresponding to the paper Diffusion Earth Mover's Distance and Distribution Embeddings
  ☆38Nov 29, 2024Updated last year
• aliutkus / swf

  View on GitHub
  ☆15Mar 30, 2020Updated 5 years ago
• bunnech / condot

  View on GitHub
  Supervised Training of Conditional Monge Maps
  ☆19Oct 30, 2023Updated 2 years ago
• chervud / AGD-vs-Sinkhorn

  View on GitHub
  Code for http://proceedings.mlr.press/v80/dvurechensky18a.html
  ☆17Jul 31, 2018Updated 7 years ago
• Daniil-Selikhanovych / ASBM

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

  View on GitHub
  Learning the optimal transport map via input convex neural neworks
  ☆42Jun 26, 2020Updated 5 years ago
• cantinilab / OT-scOmics

  View on GitHub
  This Python package will allow you to replicate the experiments from our research on applying Optimal Transport as a similarity metric in…
  ☆43Feb 3, 2023Updated 3 years ago
• milenagazdieva / LightUnbalancedOptimalTransport

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

  View on GitHub
  A set of tests for evaluating large-scale algorithms for Wasserstein-1 transport computation (NeurIPS'22).
  ☆20Sep 9, 2024Updated last year
• necludov / jam

  View on GitHub
  Implementation of Action Matching
  ☆51Jun 5, 2023Updated 2 years ago
• bunnech / cellot

  View on GitHub
  Learning Single-Cell Perturbation Responses using Neural Optimal Transport
  ☆157Oct 31, 2024Updated last year
• IBM / wasserstein-barycenters

  View on GitHub
  Code for our ICLR19 paper "Wasserstein Barycenters for Model Ensembling", Pierre Dognin, Igor Melnyk, Youssef Mroueh, Jarret Ross, Cicero…
  ☆23Nov 18, 2019Updated 6 years ago
• r-three / mats

  View on GitHub
  ☆33Jul 8, 2024Updated last year
• sbyebss / variational_wgf

  View on GitHub
  [ICML2022] Variational Wasserstein gradient flow
  ☆24Oct 17, 2022Updated 3 years ago
• willxxy / Text-EGM

  View on GitHub
  [CHIL 2024] Interpretation of Intracardiac Electrograms Through Textual Representations
  ☆12Sep 4, 2024Updated last year
• joeybose / GGM_LOG_Tutorial

  View on GitHub
  Geometric Generative Model Tutorial at LoG 2024
  ☆27Nov 27, 2024Updated last year
• ot-triton-lab / flash-sinkhorn

  View on GitHub
  FlashSinkhorn: IO-Aware Entropic Optimal Transport in PyTorch + Triton. Streaming Sinkhorn with O(nd) memory.
  ☆31Updated this week
• GiulsLu / Sinkhorn-Barycenters

  View on GitHub
  Sinkhorn Barycenters via Frank-Wolfe algorithm
  ☆27Feb 3, 2020Updated 6 years ago
• TianrongChen / DMSB

  View on GitHub
  Official implementation of Deep Momentum Schrödinger Bridge
  ☆30Nov 5, 2023Updated 2 years ago
• DoronHav / WassersteinFlowMatching

  View on GitHub
  ☆39Jan 27, 2026Updated 2 weeks ago
• GFNOrg / gfn-diffusion

  View on GitHub
  ☆36Mar 29, 2025Updated 10 months ago
• necludov / wl-mechanics

  View on GitHub
  ☆32May 30, 2024Updated last year
• antonioterpin / jkonet-star

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

  View on GitHub
  ☆34Oct 23, 2025Updated 3 months ago
• eddardd / WBTransport

  View on GitHub
  (ICASSP'21/CVPR'21) Wasserstein Barycenter Transport
  ☆32Aug 31, 2022Updated 3 years ago
• gchojnowski / gapTrick

  View on GitHub
  rebuilds and completes models of protein complexes using AlphaFold2
  ☆15Jan 22, 2026Updated 3 weeks ago
• Form2Seq-Data / Dataset

  View on GitHub
  Dataset corresponding to the paper: "Form2Seq : A Framework for Higher-Order Form Structure Extraction"
  ☆10Feb 17, 2021Updated 4 years ago
• ngushchin / EntropicNeuralOptimalTransport

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

  View on GitHub
  Mixture of Gaussian Processes Model for Sparse Longitudinal Data
  ☆38Dec 6, 2022Updated 3 years ago
• JmlrOrg / jmlr-coverletter

  View on GitHub
  JMLR Cover Letter Template
  ☆10Dec 15, 2021Updated 4 years ago