Ott-jax

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

What's Changed
* add glossary by marcocuturi in https://github.com/ott-jax/ott/pull/594
* introduce `sinkdiv` simplified API, expand glossary with sinkhorn divergence and LR sinkhorn by marcocuturi in https://github.com/ott-jax/ott/pull/596
* propagate `sinkdiv` to Monge gap implementation by marcocuturi in https://github.com/ott-jax/ott/pull/597
* add Hungarian solver from optax by marcocuturi in https://github.com/ott-jax/ott/pull/598
* Feature/refactor initializers by michalk8 in https://github.com/ott-jax/ott/pull/599
* Fix Wrong implementation of u_t(x|z) in Brownian Bridge by selmanozleyen in https://github.com/ott-jax/ott/pull/600
* Fix CI by michalk8 in https://github.com/ott-jax/ott/pull/601
* Feature/refactor relative epsilon by michalk8 in https://github.com/ott-jax/ott/pull/602
* Remove source/target masks by michalk8 in https://github.com/ott-jax/ott/pull/603


**Full Changelog**: https://github.com/ott-jax/ott/compare/0.4.9...0.5.0

What's Changed
* refactor sinkhorn_divergence by marcocuturi in https://github.com/ott-jax/ott/pull/577
* automatic scaling of `epsilon` using `std` instead of `mean` by default. by marcocuturi in https://github.com/ott-jax/ott/pull/578
* Feature/refactor was solver by michalk8 in https://github.com/ott-jax/ott/pull/580
* Fix dtype promotion in `Sinkhorn` by michalk8 in https://github.com/ott-jax/ott/pull/583
* expose `SinkhornDivergenceOutput` by marcocuturi in https://github.com/ott-jax/ott/pull/584
* Switch to `jax.random.key` by michalk8 in https://github.com/ott-jax/ott/pull/586
* fix default for intermediate quadratic layers in ICNN by marcocuturi in https://github.com/ott-jax/ott/pull/587
* Enable 3.13 CI by michalk8 in https://github.com/ott-jax/ott/pull/589
* Feature/batched vmap by michalk8 in https://github.com/ott-jax/ott/pull/588


**Full Changelog**: https://github.com/ott-jax/ott/compare/0.4.8...0.4.9

What's Changed
* Fix low-rank convergence criterion by michalk8 in https://github.com/ott-jax/ott/pull/547
* Fix/quadratic pytree by michalk8 in https://github.com/ott-jax/ott/pull/555
* Fix/gw to lr by michalk8 in https://github.com/ott-jax/ott/pull/558
* TI transport map by michalk8 in https://github.com/ott-jax/ott/pull/559
* Update tutorial structure by michalk8 in https://github.com/ott-jax/ott/pull/560
* Fix `Grid.prepare_divergences` by michalk8 in https://github.com/ott-jax/ott/pull/563
* Raise error when instatiating cost/kernel for a grid by michalk8 in https://github.com/ott-jax/ott/pull/565
* correct convergence criterion for GW by marcocuturi in https://github.com/ott-jax/ott/pull/566
* Feature/progot by michalk8 in https://github.com/ott-jax/ott/pull/567
* Fix/univariate non ti by michalk8 in https://github.com/ott-jax/ott/pull/569
* sink div for LR by marcocuturi in https://github.com/ott-jax/ott/pull/568
* Add tutorial for unbalanced OT by nvesseron in https://github.com/ott-jax/ott/pull/291
* Edits to ProgOT by pkassraie in https://github.com/ott-jax/ott/pull/570
* Feature/update progot by michalk8 in https://github.com/ott-jax/ott/pull/571
* update mm-sink tutorial and plot by zoepiran in https://github.com/ott-jax/ott/pull/572
* Feature/update prox by michalk8 in https://github.com/ott-jax/ott/pull/575
* implements sliced W by marcocuturi in https://github.com/ott-jax/ott/pull/576


**Full Changelog**: https://github.com/ott-jax/ott/compare/0.4.7...0.4.8

What's Changed
* add dropout to VelocityField and OTFM/GENOT by MUCDK in https://github.com/ott-jax/ott/pull/521
* Update icnn.py by marcocuturi in https://github.com/ott-jax/ott/pull/524
* Update Codecov Action to v4 by michalk8 in https://github.com/ott-jax/ott/pull/523
* Add h-transform for TI costs by michalk8 in https://github.com/ott-jax/ott/pull/525
* Update univariate.py by marcocuturi in https://github.com/ott-jax/ott/pull/527
* Fix/h legendre tests by michalk8 in https://github.com/ott-jax/ott/pull/529
* Update costs.py by marcocuturi in https://github.com/ott-jax/ott/pull/532
* Debiased property change by michalk8 in https://github.com/ott-jax/ott/pull/536
* Don't stop grads w.r.t. the parametrized costs by michalk8 in https://github.com/ott-jax/ott/pull/544
* univariate dual vectors for arbitrary sizes by marcocuturi in https://github.com/ott-jax/ott/pull/505
* Fix low-rank convergence criterion by michalk8 in https://github.com/ott-jax/ott/pull/547
* Fixing typos in documentation by pkassraie in https://github.com/ott-jax/ott/pull/548
* add MMSinkhorn solver by marcocuturi in https://github.com/ott-jax/ott/pull/549
* Force ICNN to adopt default initialization of its own layers by Algue-Rythme in https://github.com/ott-jax/ott/pull/551
* Deprecate Python 3.8 by michalk8 in https://github.com/ott-jax/ott/pull/554
* New neural OT solver ENOT by nazarblch in https://github.com/ott-jax/ott/pull/503
* Refactor regularized TI costs by michalk8 in https://github.com/ott-jax/ott/pull/553

New Contributors
* Algue-Rythme made their first contribution in https://github.com/ott-jax/ott/pull/551
* nazarblch made their first contribution in https://github.com/ott-jax/ott/pull/503

**Full Changelog**: https://github.com/ott-jax/ott/compare/0.4.6...0.4.7

What's Changed
* Enable M1 runner by michalk8 in https://github.com/ott-jax/ott/pull/486
* Potentials as tuple by michalk8 in https://github.com/ott-jax/ott/pull/488
* Remove `bool` option for `scale_cost` by michalk8 in https://github.com/ott-jax/ott/pull/492
* Use more iterations for LR-GW by michalk8 in https://github.com/ott-jax/ott/pull/494
* update sinhorn divergence gradient flow nb by marcocuturi in https://github.com/ott-jax/ott/pull/497
* add barycenter operator for (convex) translation invariant costs by marcocuturi in https://github.com/ott-jax/ott/pull/498
* Addition of gangbo-mccann map estimators using twist operator by marcocuturi in https://github.com/ott-jax/ott/pull/500
* Add sparse Chebyshev approximation by guillaumehu in https://github.com/ott-jax/ott/pull/502
* Fix/io callback by michalk8 in https://github.com/ott-jax/ott/pull/506
* Maint/third lower bound by michalk8 in https://github.com/ott-jax/ott/pull/507
* Remove epsilon scheduler in GW by michalk8 in https://github.com/ott-jax/ott/pull/508
* Fix SCOT tutorial imports and data by michalk8 in https://github.com/ott-jax/ott/pull/511
* restructuring neural models + addition of OT-FM and GENOT by MUCDK in https://github.com/ott-jax/ott/pull/468
* compute_sparse_laplacian gives int32 vs int64 index mismatch by selmanozleyen in https://github.com/ott-jax/ott/pull/510
* Update One_Sinkhorn.ipynb by pkassraie in https://github.com/ott-jax/ott/pull/515
* Update actions' versions by michalk8 in https://github.com/ott-jax/ott/pull/516
* Fix/batched non sym cost by michalk8 in https://github.com/ott-jax/ott/pull/518
* bug fix: avoid mixing up linear and quadratic in genot by soerenab in https://github.com/ott-jax/ott/pull/517

New Contributors
* selmanozleyen made their first contribution in https://github.com/ott-jax/ott/pull/510
* pkassraie made their first contribution in https://github.com/ott-jax/ott/pull/515
* soerenab made their first contribution in https://github.com/ott-jax/ott/pull/517

**Full Changelog**: https://github.com/ott-jax/ott/compare/0.4.5...0.4.6