GitHub repository: ott-jax/ott
Primary language: Python
Description: Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.