Skip to content

0.8.1
Compare
Choose a tag to compare
@rflamary rflamary released this 27 Dec 14:24
· 202 commits to master since this release
0.8.1
5ed6168
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

This release fixes several bugs and introduces two new backends: Cupy and Tensorflow. Note that the tensorflow backend will work only when tensorflow has enabled the Numpy behavior (for transpose that is not by default in tensorflow). We also introduce a simple benchmark on CPU GPU for the sinkhorn solver that will be provided in the backend documentation.

This release also brings a few changes in dependencies and compatibility. First we removed tests for Python 3.6 that will not be updated in the future. Also note that POT now depends on Numpy (>= 1.20) because a recent change in ABI is making the wheels non-compatible with older numpy versions. If you really need an older numpy POT will work with no problems but you will need to build it from source.

As always we want to that the contributors who helped make POT better (and bug free).

New features

  • New benchmark for sinkhorn solver on CPU/GPU and between backends (PR #316)
  • New tensorflow backend (PR #316)
  • New Cupy backend (PR #315)
  • Documentation always up-to-date with README, RELEASES, CONTRIBUTING and
    CODE_OF_CONDUCT files (PR #316, PR #322).

Closed issues

  • Fix bug in older Numpy ABI (<1.20) (Issue #308, PR #326)
  • Fix bug in ot.dist function when non euclidean distance (Issue #305, PR #306)
  • Fix gradient scaling for functions using nx.set_gradients (Issue #309, PR #310)
  • Fix bug in generalized Conditional gradient solver and SinkhornL1L2 (Issue #311, PR #313)
  • Fix log error in gromov_barycenters (Issue #317, PR #3018)
Assets 33
Loading