felix0097

felix0097's Stars

• ott-jax/ott

  Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.

  Language:Python5551019484