Skip to content

[MRG] Proximal point solver for batched optimal transport#832

Open
thibaut-germain wants to merge 17 commits into
PythonOT:masterfrom
thibaut-germain:proximal_batch_ot
Open

[MRG] Proximal point solver for batched optimal transport#832
thibaut-germain wants to merge 17 commits into
PythonOT:masterfrom
thibaut-germain:proximal_batch_ot

Conversation

@thibaut-germain

Copy link
Copy Markdown
Contributor

Types of changes

Implementation of the Bregman divergence based proximal point solver for optimal transport.

Motivation and context / Related issue

Provide differentiable approximation of exact optimal transport plans computed in parallel.

How has this been tested (if it applies)

New tests implemented in test/batch/test_solve_batch.py. Tests on all configurations possible and transport plan accuracy.

PR checklist

  • [ x] I have read the CONTRIBUTING document.
  • [ x] The documentation is up-to-date with the changes I made (check build artifacts).
  • [ x] All tests passed, and additional code has been covered with new tests.
  • [ x] I have added the PR and Issue fix to the RELEASES.md file.

@rflamary rflamary changed the title proximal point solver for batched optimal transport [WIP] proximal point solver for batched optimal transport Jul 3, 2026

@rflamary rflamary left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Neraly there @thibaut-germain could you update the example file?

Comment thread ot/batch/_linear.py Outdated
Comment thread examples/backends/plot_ot_batch.py Outdated
Comment thread examples/backends/plot_ot_batch.py Outdated
@rflamary rflamary changed the title [WIP] proximal point solver for batched optimal transport [MRG] proximal point solver for batched optimal transport Jul 6, 2026
Co-authored-by: Rémi Flamary <remi.flamary@gmail.com>
@rflamary rflamary changed the title [MRG] proximal point solver for batched optimal transport [MRG] Proximal point solver for batched optimal transport Jul 6, 2026
@codecov

codecov Bot commented Jul 6, 2026

Copy link
Copy Markdown

Codecov Report

❌ Patch coverage is 97.60000% with 3 lines in your changes missing coverage. Please review.
✅ Project coverage is 96.86%. Comparing base (c836b97) to head (f7e82ac).

Additional details and impacted files
@@           Coverage Diff           @@
##           master     #832   +/-   ##
=======================================
  Coverage   96.85%   96.86%           
=======================================
  Files         128      128           
  Lines       25593    25659   +66     
=======================================
+ Hits        24788    24854   +66     
  Misses        805      805           
🚀 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

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants