OTT-JAX
Results
1
repositories owned by
OTT-JAX
ott
462
Stars
78
Forks
Watchers
Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
OTT-JAX
Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.