OTT is a JAX toolbox that bundles a few utilities to solve optimal transport problems.
https://github.com/ott-jax/ott
#521