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

What's Changed
* Re-run and refactoring of some docs/tutorials + minor changes by marcocuturi in https://github.com/ott-jax/ott/pull/431
* Feature/quadratic solve by michalk8 in https://github.com/ott-jax/ott/pull/433
* Fix `n_iters` by michalk8 in https://github.com/ott-jax/ott/pull/437
* Fix/rank2 init by michalk8 in https://github.com/ott-jax/ott/pull/442
* nn.dataset.GaussianMixture: minor fix to the docstring and standard deviation parameter name by bamos in https://github.com/ott-jax/ott/pull/445
* introduce multivariate cdf / quantiles by marcocuturi in https://github.com/ott-jax/ott/pull/447
* Feature/update jax version by michalk8 in https://github.com/ott-jax/ott/pull/448
* Histogram Transport Implementation by Daniel-Packer in https://github.com/ott-jax/ott/pull/444
* Doc refresh by marcocuturi in https://github.com/ott-jax/ott/pull/450
* Fix/jax prng deprecation by michalk8 in https://github.com/ott-jax/ott/pull/455
* Added CDF distance to Univariate Solver by Daniel-Packer in https://github.com/ott-jax/ott/pull/451
* Use PyPI trusted publishers by michalk8 in https://github.com/ott-jax/ott/pull/459
* first lb -> third lb by Daniel-Packer in https://github.com/ott-jax/ott/pull/462
* Restructure neural module by michalk8 in https://github.com/ott-jax/ott/pull/466
* refactoring ``UnivariateSolver`` by marcocuturi in https://github.com/ott-jax/ott/pull/472
* Feature/geodesic sinkhorn by guillaumehu in https://github.com/ott-jax/ott/pull/457
* Enable 3.12 CI by michalk8 in https://github.com/ott-jax/ott/pull/475
* Feature/apply lse mode by michalk8 in https://github.com/ott-jax/ott/pull/474
* Fix/neural docs by michalk8 in https://github.com/ott-jax/ott/pull/473
* Add quadratic layers and enhance ICNNs, update tutorial by nvesseron in https://github.com/ott-jax/ott/pull/477
* Fix heat scaling by guillaumehu in https://github.com/ott-jax/ott/pull/481
* Add low-rank kernel geometry by michalk8 in https://github.com/ott-jax/ott/pull/440

New Contributors
* Daniel-Packer made their first contribution in https://github.com/ott-jax/ott/pull/444
* guillaumehu made their first contribution in https://github.com/ott-jax/ott/pull/457
* nvesseron made their first contribution in https://github.com/ott-jax/ott/pull/477

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

What's Changed
* introduce new `norm` operator that can differentiate through `lambda x: norm(x-x)` + lighter tests across the board by marcocuturi in https://github.com/ott-jax/ott/pull/411
* Feature/better io callbacks by michalk8 in https://github.com/ott-jax/ott/pull/413
* Feature/optional dependencies by michalk8 in https://github.com/ott-jax/ott/pull/412
* Update pre-commits by michalk8 in https://github.com/ott-jax/ott/pull/416
* Feature/move solve by michalk8 in https://github.com/ott-jax/ott/pull/418
* Feature/lineax ridge by michalk8 in https://github.com/ott-jax/ott/pull/424
* Feature/ulrgw by michalk8 in https://github.com/ott-jax/ott/pull/410


**Full Changelog**: https://github.com/ott-jax/ott/compare/0.4.3...0.4.4

What's Changed
* A few changes to increase Coverage by marcocuturi in https://github.com/ott-jax/ott/pull/390
* rename `ent_reg_cost` in sinkhorn solver by marcocuturi in https://github.com/ott-jax/ott/pull/391
* Increase coverage of NN modules by including init in tests by marcocuturi in https://github.com/ott-jax/ott/pull/392
* add tests for plots in ``potentials`` by marcocuturi in https://github.com/ott-jax/ott/pull/393
* test `Plot` class in `tools` module to increase coverage by marcocuturi in https://github.com/ott-jax/ott/pull/394
* Add soft `topk_mask` operator, fix `ranks` docs and tests by marcocuturi in https://github.com/ott-jax/ott/pull/396
* Add `relative_epsilon` option to `GromovWasserstein` by michalk8 in https://github.com/ott-jax/ott/pull/355
* [ENH]: Fix misc typos in soft_sort doc by pierreablin in https://github.com/ott-jax/ott/pull/397
* Update docs by michalk8 in https://github.com/ott-jax/ott/pull/398
* add ability to add titles on animation plots by marcocuturi in https://github.com/ott-jax/ott/pull/400
* Update contributing visibility by michalk8 in https://github.com/ott-jax/ott/pull/399
* update nb by marcocuturi in https://github.com/ott-jax/ott/pull/403
* Fix not being able to jit the soft quantile by michalk8 in https://github.com/ott-jax/ott/pull/408
* Test `default_progress_fn`, skip `types.py` by michalk8 in https://github.com/ott-jax/ott/pull/404


**Full Changelog**: https://github.com/ott-jax/ott/compare/0.4.2...0.4.3

What's Changed
* Add a tuto about sparse monge displacements by pierreablin in https://github.com/ott-jax/ott/pull/374
* introduce `ent_reg_cost` / `kl_reg_cost` in Sinkhorn to clarify which is returned to the user in `SinkhornOutput` objects. by marcocuturi in https://github.com/ott-jax/ott/pull/376
* more precise pydocs for 376 by marcocuturi in https://github.com/ott-jax/ott/pull/377
* Update sinkhorn.py by marcocuturi in https://github.com/ott-jax/ott/pull/378
* Compute multiple soft-quantiles in one execution without using `vmap` by marcocuturi in https://github.com/ott-jax/ott/pull/382
* incorporate feedback in 382 by marcocuturi in https://github.com/ott-jax/ott/pull/385
* Update soft_sort.py by marcocuturi in https://github.com/ott-jax/ott/pull/387
* Add elastic-net type costs with an orthogonal linear transformation of the input before it is fed into the regularizer by michalk8 in https://github.com/ott-jax/ott/pull/383
* Add monge gap by theouscidda6 in https://github.com/ott-jax/ott/pull/361
* Feature/unbalanced LR sinkhorn by michalk8 in https://github.com/ott-jax/ott/pull/379
* changes in definition of Monge gap by marcocuturi in https://github.com/ott-jax/ott/pull/389
* fix 375, `converged` flag when predefined number of iterations. by marcocuturi in https://github.com/ott-jax/ott/pull/386


**Full Changelog**: https://github.com/ott-jax/ott/compare/0.4.1...0.4.2