Optimal Transport tools implemented with the JAX framework, to get auto-diff, parallel and jit-able computations.
Latest commits.
Builders behind this project.
