Toolbox to integrate optimal transport loss functions using automatic differentiation and Sinkhorn's algorithm
☆450May 14, 2018Updated 7 years ago
Alternatives and similar repositories for SinkhornAutoDiff
Users that are interested in SinkhornAutoDiff are comparing it to the libraries listed below
Sorting:
- Approximating Wasserstein distances with PyTorch☆458Apr 15, 2023Updated 2 years ago
- Geometric loss functions between point clouds, images and volumes☆683Apr 14, 2025Updated 10 months ago
- POT : Python Optimal Transport☆2,757Feb 24, 2026Updated last week
- Keras implementation of Deep Wasserstein Embeddings☆48Apr 15, 2018Updated 7 years ago
- implements optimal transport algorithms in pytorch☆105May 15, 2022Updated 3 years ago
- Totally Versatile Miscellanea for Pytorch☆477Mar 20, 2022Updated 3 years ago
- MMD, Hausdorff and Sinkhorn divergences scaled up to 1,000,000 samples.☆58Apr 15, 2019Updated 6 years ago
- Stochastic algorithms for computing Regularized Optimal Transport☆58Aug 16, 2018Updated 7 years ago
- Stochastic Optimization for Optimal Transport☆22Oct 27, 2016Updated 9 years ago
- Tensorflow implementation of optimal transport (OT) with Sinkhorn algorithm.☆19Nov 5, 2019Updated 6 years ago
- A tensorflow implementation of the Sinkhorn Distance calculation☆27Apr 28, 2019Updated 6 years ago
- Code accompanying our paper at AISTATS 2020☆21Jan 12, 2021Updated 5 years ago
- Code for Deep Parametric Shape Predictions using Distance Fields, in CVPR 2020.☆54Mar 28, 2021Updated 4 years ago
- ☆15Sep 5, 2016Updated 9 years ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,159Feb 6, 2026Updated 3 weeks ago
- Optimal Transport for Dummies - Code, slides and article☆33May 29, 2017Updated 8 years ago
- Implementation of the Sliced Wasserstein Autoencoder using PyTorch☆101Sep 24, 2018Updated 7 years ago
- PyTorch Framework for Developing Memory Efficient Deep Invertible Networks☆255May 10, 2023Updated 2 years ago
- TensorFlow implementation of (Momentum) Stochastic Variance-Adapted Gradient.☆44May 11, 2018Updated 7 years ago
- code for paper "Improving Sequence-to-Sequence Learning via Optimal Transport"☆68Jun 24, 2019Updated 6 years ago
- Torch modules that wrap blackbox combinatorial solvers according to the method presented in "Differentiating Blackbox Combinatorial Solve…☆348Dec 21, 2021Updated 4 years ago
- Annealed Importance Sampling (AIS) for generative models.☆16Jul 20, 2018Updated 7 years ago
- ☆172Oct 8, 2022Updated 3 years ago
- Gabriel Peyré, Marco Cuturi, Justin Solomon, Gromov-Wasserstein Averaging of Kernel and Distance Matrices, Proc. of ICML 2016.☆75Aug 21, 2016Updated 9 years ago
- Learning generative models with Sinkhorn Loss☆30Nov 9, 2018Updated 7 years ago
- ☆24Sep 10, 2021Updated 4 years ago
- A list of awesome papers and cool resources on optimal transport and its applications in general! As you will notice, this list is curren…☆246May 9, 2021Updated 4 years ago
- A python implementation of discrete optimal transport with a Tsallis entropy regularization.☆14Oct 23, 2023Updated 2 years ago
- Differentiable convex optimization layers☆2,060Updated this week
- Implementation of the Sliced Wasserstein Autoencoders☆92Jun 6, 2018Updated 7 years ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,627Mar 25, 2022Updated 3 years ago
- It is a repo which allows to compute all divergences derived from the theory of entropically regularized, unbalanced optimal transport. I…☆31Dec 21, 2022Updated 3 years ago
- Pytorch Implemetation for our NAACL2019 Paper "Riemannian Normalizing Flow on Variational Wasserstein Autoencoder for Text Modeling" http…☆63Apr 1, 2020Updated 5 years ago
- A fast and differentiable QP solver for PyTorch.☆784Sep 3, 2024Updated last year
- Source code for the ICML2019 paper "Subspace Robust Wasserstein Distances"☆29Jun 14, 2019Updated 6 years ago
- [NeurIPS'19] Deep Equilibrium Models☆793Jul 4, 2022Updated 3 years ago
- Code for Optimal Transport for structured data with application on graphs☆104Dec 17, 2022Updated 3 years ago
- ☆10Jun 22, 2022Updated 3 years ago
- Morgan A. Schmitz., Matthieu Heitz, Nicolas Bonneel, Fred Ngole, David Coeurjolly, Marco Cuturi, Gabriel Peyré, and Jean-Luc Starck. "Was…☆20Oct 18, 2019Updated 6 years ago