Skip to content

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

Merged
rflamary merged 17 commits into
PythonOT:masterfrom
thibaut-germain:proximal_batch_ot
Jul 7, 2026
Merged

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

Conversation

@thibaut-germain

@thibaut-germainthibaut-germain commented Jul 3, 2026

Copy link
Copy Markdown
Contributor

Types of changes

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

Fixes#532 and #482

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.

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

@rflamaryrflamary 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 threadot/batch/_linear.py Outdated
Comment threadexamples/backends/plot_ot_batch.py Outdated
Comment threadexamples/backends/plot_ot_batch.py Outdated
@rflamaryrflamary changed the title [WIP] proximal point solver for batched optimal transport[MRG] proximal point solver for batched optimal transportJul 6, 2026
Co-authored-by: Rémi Flamary <remi.flamary@gmail.com>
@rflamaryrflamary changed the title [MRG] proximal point solver for batched optimal transport[MRG] Proximal point solver for batched optimal transportJul 6, 2026
@codecov

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

@rflamary
rflamary merged commit e52c2a3 into PythonOT:masterJul 7, 2026
21 checks passed
Flastre pushed a commit to Flastre/POT that referenced this pull request Jul 8, 2026
* correction SGOT cost matrix, added to contributor list, move sgot example to other
* updated graphs
* fix plots
* update example
* fix releases.md
* reformating
* updating format
* updated format
* updated format
* added PR and references
* aligned usage of solve and solve_batch + added batch tests
* fixed example
* Apply suggestions from code review
Co-authored-by: Rémi Flamary <remi.flamary@gmail.com>
---------
Co-authored-by: Sienna O'Shea <osheasienna@gmail.com>
Co-authored-by: Rémi Flamary <remi.flamary@gmail.com>
Sign up for freeto 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.

Can not batch ot.emd2 via torch.vmap

3 participants

@thibaut-germain@rflamary@osheasienna