Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
created at Dec. 24, 2021, 7:28 a.m.
10 +0
520 +4
80 +0