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_iters by @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 UnivariateSolver by @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