ott-jax/ott

Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.