bamos / ott

Optimal Transport tools implemented with the JAX framework, to get auto-diff, parallel and jit-able computations.

https://ott-jax.readthedocs.io

Geek Repo

Github PK Tool