Releases: ott-jax/ott
Releases · ott-jax/ott
Minor release: Switching back to Python 3.7
Switching back to Python 3.7 to avoid bug with colab.
Version 2.0
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 aSinkhorn
solver class, usingSinkhornState
andSinkhornOutput
variables. In this refactoring,momentum
andanderson
are now objects. - Same for
gromov_wasserstein
which now runs solverGromovWasserstein
- OT problems are now defined in
problems
andquad_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...