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
456 +0
79 +0