POT : Python Optimal Transport
Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.
Fast Sampled Gromov-Wasserstein optimal transport solver — pure PyTorch, scalable, differentiable