Skip to content

Releases: ott-jax/ott

0.4.0: GPU tests, re-adding jit, improved NN OT solvers, and misc.

13 Feb 10:09
40bf4af
Compare
Choose a tag to compare

What's Changed

New Contributors

Full Changelog: 0.3.1...0.4.0

0.3.1 with new cost features, different code layout, changes in potentials handling. Some solvers in function form deprecated.

15 Dec 22:53
2f5242b
Compare
Choose a tag to compare

What's Changed

Full Changelog: 0.3.0...0.3.1

New organization for code; Changes in `costs` for `PointClouds`

23 Nov 09:37
156ef2b
Compare
Choose a tag to compare

The main changes in this version are twofold:

  • Changes in PointCloud geometry, and more specifically handling of cost function. The power parameter that was used to optionally pass a c(x,y) = CostFn(x,y) ** power is now deprecated. One can add it manually by defining a custom CostFn. To compensate this change, a new class of translation invariant costs (TICost) has been created, from which most costs now inherit, defined as c(x,y) = h(x-y). Additionally, to handle Brenier's theorem, the user has the option of passing on the legendre transform of h, h_legendre.
  • The core folder was too horizontal, containing various modules. It has been reorganized and split into 3 modules that make more sense, problems (to describe OT problems), solvers (solve them) and initializers (as optional modules to help solvers). The latter two have an arborescence that reflects that in problems.

What's Changed

Full Changelog: 0.2.11...0.3.0

adaptive momentum by default, Euclidean is now SqEuclidean (W2), Euclidean is for W1 and uses `jnp.linalg.norm` to ensure differentiability

19 Oct 22:43
8646692
Compare
Choose a tag to compare

What's Changed

Full Changelog: 0.2.10...0.2.11

Fix various bugs and add Meta initializer

14 Oct 08:59
fb778fc
Compare
Choose a tag to compare

What's Changed

New Contributors

Full Changelog: 0.2.9...0.2.10

0.2.9: new initializations for Sinkhorn, LR Sinkhorn and LR GW (including k-means++ implementations), entropic map, faster Sinkhorn implicit diff

03 Oct 10:17
7e7d7b6
Compare
Choose a tag to compare

What's Changed

Full Changelog: 0.2.8...0.2.9

Extension of barycenters to (Fused) GW, several bug fixes around LR Sinkhorn, parallel (segment) code for Sinkhorn, inits for ICNNs

03 Aug 14:53
22d3929
Compare
Choose a tag to compare

What's Changed

Full Changelog: 0.2.7...0.2.8

Fixing continuous barycenter, cleaning up docs and more generally the packaging of OTT.

22 Jun 14:58
d212d40
Compare
Choose a tag to compare

What's Changed

New Contributors

Full Changelog: 0.2.6...0.2.7

Quite a few bug fixes on jitting / scaling of costs / convergence

24 May 15:33
27e59c4
Compare
Choose a tag to compare

What's Changed

New Contributors

Full Changelog: 0.2.5...0.2.6

Bug in not_converged flag that was impacting convergence fixed.

02 Apr 07:51
5a86d14
Compare
Choose a tag to compare

What's Changed

New Contributors

Full Changelog: 0.2.4...0.2.5