0.4.1 Various new features (kernel steps for LR Sinkhon, lineax implicit diff), solvers must now be jitted.

Highlights:

• reformatted tutorial NBs
• progressbars for Sinkhorn
• low-rank Sinkhorn solvers can run in kernel mode
• discrete_barycenter is refactored using the Solver/Problem model, rather than a simple function.
• soft-DTW added as a cost function.
• Sparsity inducing (in displacements) costs, see https://arxiv.org/abs/2302.04065

What's Changed

• Fix typo: Wasserstein-4 -> Wasserstein-2 by @bamos in #249
• Fix/warnings lint by @michalk8 in #260
• Addresses #229: Copyrights unification by @AWehenkel in #261
• fixed the docs about the implicit_diff argument by @gjhuizing in #270
• Correct package requirements for tests. by @AWehenkel in #271
• Add pre-commit CI cache by @michalk8 in #279
• Fix flake8 D10{1,2,3} errors by @giovp in #269
• Added CostFn and Plot to the documentation by @gjhuizing in #288
• add primal_cost property to output of GW, to compute cost without any regularization. by @marcocuturi in #286
• add ruff and remove redundant pre-commits by @giovp in #278
• Remove matplotlib dependency by @michalk8 in #304
• Fix/epsilon regularization by @michalk8 in #310
• Address #252 - Make soft ranking notebook self contained by @AWehenkel in #301
• Fix small TODOs by @michalk8 in #311
• Add CI tests for upstream JAX by @michalk8 in #313
• Subsample initializer by @JTT94 in #309
• feat: add progress reporting to Sinkhorn by @bosr in #228
• Refactor discrete_barycenter as solver, cf. issue #255 by @marcocuturi in #316
• Make random seeding consistent with PRNG keys everywhere instead of seeds by @othmanesebbouh in #290
• Add Soft-DTW cost function by @michalk8 in #318
• Fix/sinkdiv hessian by @michalk8 in #321
• Enable D* checks by @michalk8 in #319
• Fix spelling by @michalk8 in #323
• Remove ignoring passed params in ICNN by @michalk8 in #325
• Fix spelling by @michalk8 in #326
• neuraldual: fix issues with objective-based amortization and no-fine-tuning by @bamos in #327
• Parameter to make line strength proportional to coupling strength in Plot by @gjhuizing in #280
• Remove sparse option in Graph geometry by @michalk8 in #329
• gh: refactor community files by @SauravMaheshkar in #331
• Have a simpler "getting started" notebook by @pierreablin in #296
• update NBs by @marcocuturi in #332
• Docs/fix minor issues by @michalk8 in #334
• Small fix in soft_sort.quantile docstring by @pierreablin in #336
• Fix/jit inside jit by @michalk8 in #335
• typos by @marcocuturi in #339
• Remove assertions in cost functions by @michalk8 in #340
• Fix EntropicPotentials's test by @michalk8 in #343
• Added tutorial on Sinkhorn divergence by @gjhuizing in #282
• Pin chex version in GPU CI by @michalk8 in #346
• Feature/unscaled gw cost by @michalk8 in #348
• Fix typos in notebook by @michalk8 in #350
• docs/progress bar by @bosr in #347
• Fix/default prng by @michalk8 in #353
• tracking progress of gromov-wasserstein by @bosr in #351
• Store individual costs to in the GW barycenter by @michalk8 in #354
• Update brain notebook by @michalk8 in #358
• add dtype to kernel by @MUCDK in #365
• Neural OT notebook: add more 2d datasets by @bamos in #303
• New kernel lr by @meyerscetbon in #324
• Fine grained control of Bures barycenters by @marcocuturi in #366
• use lineax to solve linear system in implicit diff by @marcocuturi in #370
• Fix .readthedocs.yaml by @michalk8 in #372

New Contributors

• @AWehenkel made their first contribution in #261
• @gjhuizing made their first contribution in #270
• @giovp made their first contribution in #269
• @bosr made their first contribution in #228
• @othmanesebbouh made their first contribution in #290
• @pierreablin made their first contribution in #296

Full Changelog: 0.4.0...0.4.1