awehenkel / ott Goto Github PK
View Code? Open in Web Editor NEWThis project forked from ott-jax/ott
Optimal Transport tools implemented with the JAX framework, to get auto-diff, parallel and jit-able computations.
Home Page: https://ott-jax.readthedocs.io
License: Apache License 2.0