Pinned Repositories
ott
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
SOADML
Scripts for optimization class
WassersteinInverseGPU
Computation of the derivative of the Loss ( WBarycenter(Dictionary,lambda), q)
optimaltransport.github.io
Web site of the Computational Optimal Transport book
optimaltransport.tutorial.github.io
ott
Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
marcocuturi's Repositories
marcocuturi/SOADML
Scripts for optimization class
marcocuturi/WassersteinInverseGPU
Computation of the derivative of the Loss ( WBarycenter(Dictionary,lambda), q)