Skip to content

Releases: ott-jax/ott

Minor release: Switching back to Python 3.7

03 Feb 14:46
36bcf43
Compare
Choose a tag to compare

Switching back to Python 3.7 to avoid bug with colab.

Version 2.0

24 Jan 18:24
4c2225e
Compare
Choose a tag to compare

Version 2.0 is out with major changes:

  • Toolbox will now be developped in the ott-jax org's repo, and has moved out of google_research. pip path remains the same, i.e. pip install ott-jax.
  • Introduction of LRCGeometry, i.e. low rank geometries that can apply cost in efficient manner.
  • Rewritten the sinkhorn function, which is now a wrapper around a Sinkhorn solver class, using SinkhornState and SinkhornOutput variables. In this refactoring, momentum and anderson are now objects.
  • Same for gromov_wasserstein which now runs solver GromovWasserstein
  • OT problems are now defined in problems and quad_problems
  • Addition of low-rank solvers, both to solve Sinkhorn (LRSinkhorn) and to solve Gromov-Wasserstein (just need to specify a rank).
  • Unbalanced GW solver
  • Tools repository now holds tools to define and manipulate Gaussians and mixtures of Gaussians.
  • and many more...