Skip to content

Add Annealed Sinkhorn tutorial notebook, based on Chizat 2024#682

Open
elyanrgn wants to merge 2 commits into
ott-jax:mainfrom
elyanrgn:tutorial-annealed-sinkhorn
Open

Add Annealed Sinkhorn tutorial notebook, based on Chizat 2024#682
elyanrgn wants to merge 2 commits into
ott-jax:mainfrom
elyanrgn:tutorial-annealed-sinkhorn

Conversation

@elyanrgn
Copy link
Copy Markdown

@elyanrgn elyanrgn commented May 2, 2026

Summary

This PR adds a new tutorial notebook on annealed Sinkhorn under docs/tutorials/linear/.

The notebook introduces entropy-regularized optimal transport and shows how annealing the regularization parameter can improve the speed-accuracy trade-off in Sinkhorn iterations. It includes:

  • fixed-ε Sinkhorn baselines and Pareto-front comparisons,
  • geometric schedules using OTT-JAX's native Epsilon scheduler,
  • polynomial schedules motivated by convergence theory,
  • an OTT-JAX adaptation of a debiased annealed variant.
  • an experience on convergence conditions.

Notes

  • The debiased section is presented as an OTT-JAX adaptation rather than a strict reproduction of the reference algorithm.
  • The content focuses on practical intuition, implementation details, and empirical comparisons.

Submitted as part of the OTT-ENSAE 2025–2026 academic assignment.
Rougon Elyan

@review-notebook-app
Copy link
Copy Markdown

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@marcocuturi
Copy link
Copy Markdown
Contributor

marcocuturi commented May 24, 2026

excellent work, @michalk8 we may want to take a look

@codecov
Copy link
Copy Markdown

codecov Bot commented May 24, 2026

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 87.34%. Comparing base (e7d02d0) to head (7a11454).
⚠️ Report is 5 commits behind head on main.

Additional details and impacted files

Impacted file tree graph

@@            Coverage Diff             @@
##             main     #682      +/-   ##
==========================================
- Coverage   87.37%   87.34%   -0.03%     
==========================================
  Files          82       82              
  Lines        8474     8474              
  Branches      580      580              
==========================================
- Hits         7404     7402       -2     
- Misses        921      922       +1     
- Partials      149      150       +1     

see 1 file with indirect coverage changes

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants