Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
updated at Nov. 1, 2024, 12:44 p.m.
10 +0
520 +4
80 +0