Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
updated at May 17, 2024, 6:39 p.m.
10 +0
458 +0
78 +0