https://github.com/ott-jax/ott

Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.
https://github.com/ott-jax/ott

automatic-differentiation gromov-wasserstein jax optimal-transport sinkhorn

Last synced: about 2 months ago
JSON representation

Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.

• Host: GitHub
• URL: https://github.com/ott-jax/ott
• Owner: ott-jax
• License: apache-2.0
• Created: 2021-12-24T07:28:56.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2025-04-22T16:39:23.000Z (2 months ago)
• Last Synced: 2025-04-26T08:07:04.158Z (2 months ago)
• Topics: automatic-differentiation, gromov-wasserstein, jax, optimal-transport, sinkhorn
• Language: Python
• Homepage: https://ott-jax.readthedocs.io
• Size: 164 MB
• Stars: 580
• Watchers: 10
• Forks: 93
• Open Issues: 50
• Metadata Files:
  • Readme: README.md
  • Contributing: CONTRIBUTING.md
  • License: LICENSE
  • Code of conduct: CODE_OF_CONDUCT.md
  • Citation: CITATION.bib

Awesome Lists containing this project

• awesome-sciml - ott-jax/ott
• awesome-jax - OTT-JAX - Optimal transport tools in JAX. <img src="https://img.shields.io/github/stars/ott-jax/ott?style=social" align="center"> (Libraries / New Libraries)
• awesome-jax - ott - Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations. <img src="https://img.shields.io/github/stars/ott-jax/ott?style=social" align="center"> (Libraries)
• awesome-jax - ott - Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations. <img src="https://img.shields.io/github/stars/ott-jax/ott?style=social" align="center"> (Libraries)