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

Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
https://github.com/ott-jax/ott

automatic-differentiation gromov-wasserstein jax optimal-transport sinkhorn

Last synced: 3 months ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/ott-jax/ott
• Owner: ott-jax
• License: apache-2.0
• Created: 2021-12-24T07:28:56.000Z (almost 3 years ago)
• Default Branch: main
• Last Pushed: 2024-04-25T14:52:47.000Z (7 months ago)
• Last Synced: 2024-04-25T20:55:52.398Z (7 months ago)
• Topics: automatic-differentiation, gromov-wasserstein, jax, optimal-transport, sinkhorn
• Language: Python
• Homepage: https://ott-jax.readthedocs.io
• Size: 76.6 MB
• Stars: 456
• Watchers: 10
• Forks: 66
• Open Issues: 69
• 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)