Releases: ott-jax/ott
Releases · ott-jax/ott
0.6.0
What's Changed
- add illustration BB nb + glossary by @marcocuturi in #662
- various docs improvements + FreeWassBaryOutput by @marcocuturi in #664
- Flow matching by @michalk8 in #663
- make sure last layer of MLP velocity is purely linear by @marcocuturi in #665
- Better final layer handling for MLP by @michalk8 in #666
- Feature/sdot to dataset by @michalk8 in #667
- new default SD + tutorial for FM including coupling dataloaders by @marcocuturi in #668
Full Changelog: 0.5.2...0.6.0
0.5.2
0.5.1
What's Changed
- Maint/fix tests by @michalk8 in #609
- solve tests for soft-sort / unb. Bures + add xlim/ylim options in plot by @marcocuturi in #610
- Fix CI by @michalk8 in #611
- Add OTCP by @michalk8 in #612
- add
Dotpcost and corresponding test by @marcocuturi in #615 - parameterize sampler for
OTCPas fn by @marcocuturi in #617 - Adds
entropyof coupling computations and instantiatesDotptoLRCGeometryby @marcocuturi in #619 - Fix: replace jnp.product with jnp.prod in grid tutorial by @zakil-02 in #622
- simplify _interpolate for progOT by @zsteve in #650
- OTCP tutorial by @EugeneNdiaye in #613
- Modify
FenchelConjugateLBFGSto handle data in higher dimensions by @donlap in #652 - replace jaxopt by optax's l-bfgs, add legendre convenience wrapper by @marcocuturi in #655
- Sharding demo by @marcocuturi in #658
New Contributors
- @zakil-02 made their first contribution in #622
- @zsteve made their first contribution in #650
- @EugeneNdiaye made their first contribution in #613
- @donlap made their first contribution in #652
Full Changelog: 0.5.0...0.5.1
0.5.0
What's Changed
- add glossary by @marcocuturi in #594
- introduce
sinkdivsimplified API, expand glossary with sinkhorn divergence and LR sinkhorn by @marcocuturi in #596 - propagate
sinkdivto Monge gap implementation by @marcocuturi in #597 - add Hungarian solver from optax by @marcocuturi in #598
- Feature/refactor initializers by @michalk8 in #599
- Fix Wrong implementation of u_t(x|z) in Brownian Bridge by @selmanozleyen in #600
- Fix CI by @michalk8 in #601
- Feature/refactor relative epsilon by @michalk8 in #602
- Remove source/target masks by @michalk8 in #603
Full Changelog: 0.4.9...0.5.0
0.4.9
What's Changed
- refactor sinkhorn_divergence by @marcocuturi in #577
- automatic scaling of
epsilonusingstdinstead ofmeanby default. by @marcocuturi in #578 - Feature/refactor was solver by @michalk8 in #580
- Fix dtype promotion in
Sinkhornby @michalk8 in #583 - expose
SinkhornDivergenceOutputby @marcocuturi in #584 - Switch to
jax.random.keyby @michalk8 in #586 - fix default for intermediate quadratic layers in ICNN by @marcocuturi in #587
- Enable 3.13 CI by @michalk8 in #589
- Feature/batched vmap by @michalk8 in #588
Full Changelog: 0.4.8...0.4.9
OTT 0.4.8
What's Changed
- Fix low-rank convergence criterion by @michalk8 in #547
- Fix/quadratic pytree by @michalk8 in #555
- Fix/gw to lr by @michalk8 in #558
- TI transport map by @michalk8 in #559
- Update tutorial structure by @michalk8 in #560
- Fix
Grid.prepare_divergencesby @michalk8 in #563 - Raise error when instatiating cost/kernel for a grid by @michalk8 in #565
- correct convergence criterion for GW by @marcocuturi in #566
- Feature/progot by @michalk8 in #567
- Fix/univariate non ti by @michalk8 in #569
- sink div for LR by @marcocuturi in #568
- Add tutorial for unbalanced OT by @nvesseron in #291
- Edits to ProgOT by @pkassraie in #570
- Feature/update progot by @michalk8 in #571
- update mm-sink tutorial and plot by @zoepiran in #572
- Feature/update prox by @michalk8 in #575
- implements sliced W by @marcocuturi in #576
Full Changelog: 0.4.7...0.4.8
OTT 0.4.7
What's Changed
- add dropout to VelocityField and OTFM/GENOT by @MUCDK in #521
- Update icnn.py by @marcocuturi in #524
- Update Codecov Action to v4 by @michalk8 in #523
- Add h-transform for TI costs by @michalk8 in #525
- Update univariate.py by @marcocuturi in #527
- Fix/h legendre tests by @michalk8 in #529
- Update costs.py by @marcocuturi in #532
- Debiased property change by @michalk8 in #536
- Don't stop grads w.r.t. the parametrized costs by @michalk8 in #544
- univariate dual vectors for arbitrary sizes by @marcocuturi in #505
- Fix low-rank convergence criterion by @michalk8 in #547
- Fixing typos in documentation by @pkassraie in #548
- add MMSinkhorn solver by @marcocuturi in #549
- Force ICNN to adopt default initialization of its own layers by @Algue-Rythme in #551
- Deprecate Python 3.8 by @michalk8 in #554
- New neural OT solver ENOT by @nazarblch in #503
- Refactor regularized TI costs by @michalk8 in #553
New Contributors
- @Algue-Rythme made their first contribution in #551
- @nazarblch made their first contribution in #503
Full Changelog: 0.4.6...0.4.7
OTT 0.4.6
What's Changed
- Enable M1 runner by @michalk8 in #486
- Potentials as tuple by @michalk8 in #488
- Remove
booloption forscale_costby @michalk8 in #492 - Use more iterations for LR-GW by @michalk8 in #494
- update sinhorn divergence gradient flow nb by @marcocuturi in #497
- add barycenter operator for (convex) translation invariant costs by @marcocuturi in #498
- Addition of gangbo-mccann map estimators using twist operator by @marcocuturi in #500
- Add sparse Chebyshev approximation by @guillaumehu in #502
- Fix/io callback by @michalk8 in #506
- Maint/third lower bound by @michalk8 in #507
- Remove epsilon scheduler in GW by @michalk8 in #508
- Fix SCOT tutorial imports and data by @michalk8 in #511
- restructuring neural models + addition of OT-FM and GENOT by @MUCDK in #468
- compute_sparse_laplacian gives int32 vs int64 index mismatch by @selmanozleyen in #510
- Update One_Sinkhorn.ipynb by @pkassraie in #515
- Update actions' versions by @michalk8 in #516
- Fix/batched non sym cost by @michalk8 in #518
- bug fix: avoid mixing up linear and quadratic in genot by @soerenab in #517
New Contributors
- @selmanozleyen made their first contribution in #510
- @pkassraie made their first contribution in #515
- @soerenab made their first contribution in #517
Full Changelog: 0.4.5...0.4.6
Neural module refactoring
What's Changed
- Re-run and refactoring of some docs/tutorials + minor changes by @marcocuturi in #431
- Feature/quadratic solve by @michalk8 in #433
- Fix
n_itersby @michalk8 in #437 - Fix/rank2 init by @michalk8 in #442
- nn.dataset.GaussianMixture: minor fix to the docstring and standard deviation parameter name by @bamos in #445
- introduce multivariate cdf / quantiles by @marcocuturi in #447
- Feature/update jax version by @michalk8 in #448
- Histogram Transport Implementation by @Daniel-Packer in #444
- Doc refresh by @marcocuturi in #450
- Fix/jax prng deprecation by @michalk8 in #455
- Added CDF distance to Univariate Solver by @Daniel-Packer in #451
- Use PyPI trusted publishers by @michalk8 in #459
- first lb -> third lb by @Daniel-Packer in #462
- Restructure neural module by @michalk8 in #466
- refactoring
UnivariateSolverby @marcocuturi in #472 - Feature/geodesic sinkhorn by @guillaumehu in #457
- Enable 3.12 CI by @michalk8 in #475
- Feature/apply lse mode by @michalk8 in #474
- Fix/neural docs by @michalk8 in #473
- Add quadratic layers and enhance ICNNs, update tutorial by @nvesseron in #477
- Fix heat scaling by @guillaumehu in #481
- Add low-rank kernel geometry by @michalk8 in #440
New Contributors
- @Daniel-Packer made their first contribution in #444
- @guillaumehu made their first contribution in #457
- @nvesseron made their first contribution in #477
Full Changelog: 0.4.4...0.4.5
Unbalanced low-rank Gromov-Wasserstein
What's Changed
- introduce new
normoperator that can differentiate throughlambda x: norm(x-x)+ lighter tests across the board by @marcocuturi in #411 - Feature/better io callbacks by @michalk8 in #413
- Feature/optional dependencies by @michalk8 in #412
- Update pre-commits by @michalk8 in #416
- Feature/move solve by @michalk8 in #418
- Feature/lineax ridge by @michalk8 in #424
- Feature/ulrgw by @michalk8 in #410
Full Changelog: 0.4.3...0.4.4