OTT-JAX

Pinned Repositories

ott
Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
Language:Python521 10 19280

OTT-JAX's Repositories

ott-jax/ott
Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
Language:Python521 10 19280