ott
Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
506 stars
10 watching
80 forks
Language: Python
last commit: 3 days ago
Linked from 1 awesome list
automatic-differentiationgromov-wassersteinjaxoptimal-transportsinkhorn
n2cholas/awesome-jax