Wasserstein (free) barycenters + normalizations + alternative initialization for Sinkhorn LR + neural dual solvers

What's Changed

• Fixes rank and epsilon arguments for LR-GW (see issue #21) by @LaetitiaPapaxanthos in #28
• Add Neural Dual Solver and Utilities by @bunnech in #32
• Migrate away from using JaxTestCase in tests by @LaetitiaPapaxanthos in #35
• Adding scaling factor to the cost matrix. by @LaetitiaPapaxanthos in #38
• Initialization LRSinkhorn by @meyerscetbon in #47
• Adding max_cost scaling for LR. by @LaetitiaPapaxanthos in #48
• Free support Wasserstein barycenter by @marcocuturi in #49

New Contributors

• @bunnech made their first contribution in #32
• @meyerscetbon made their first contribution in #39

Full Changelog: 0.2.3...0.2.4

Improved online mode, entropic regularization for LR-GW, bug fixes in GW, fixing tests

What's Changed

• Update direct transport plan computation by @theouscidda6 in #5
• Updating GW notebook to work with v0.2.2, adding unbalanced and fused GW options to Transport. by @LaetitiaPapaxanthos in #4
• Initialising linear_convergence and costs in GWState to -1 by @LaetitiaPapaxanthos in #6
• README example: transport.Transport -> transport.solve by @bamos in #7
• Fixes bug in transport.solve by @LaetitiaPapaxanthos in #8
• Remove custom vdot implementation by @michalk8 in #10
• Fixes imports in tests. by @LaetitiaPapaxanthos in #11
• Fixes tests. by @LaetitiaPapaxanthos in #12
• Fixes tests. by @LaetitiaPapaxanthos in #13
• Fixing tests. by @LaetitiaPapaxanthos in #14
• Entropic regularization for LR-GW by @marcocuturi in #19
• Batch apply_lse_kernel for online=True by @michalk8 in #23

New Contributors

• @theouscidda6 made their first contribution in #5
• @LaetitiaPapaxanthos made their first contribution in #4
• @bamos made their first contribution in #7
• @michalk8 made their first contribution in #10

Full Changelog: 0.2.2...0.2.3

Minor release: Removing icnn from __init__.

Preventing bug in colab when importing ott, as flax is not a preinstalled package.

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 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...