Skip to content

[MRG] Add a PyTorch solver to ot.dr.wda (#806) #858

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
deeb01 wants to merge 12 commits into
base: master
Choose a base branch
from
Open

[MRG] Add a PyTorch solver to ot.dr.wda (#806) #858

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
6ffce16
Add a PyTorch solver to ot.dr.wda (#806)
deeb01 Sep 14, 2026
227d5f6
Correct PR number in RELEASES.md
deeb01 Sep 15, 2026
9dc1573
Merge branch 'master' into feat-806-wda-torch-solver
rflamary Sep 15, 2026
4a3741b
Merge branch 'master' into feat-806-wda-torch-solver
rflamary Sep 15, 2026
522802a
Fix three differences between the torch and autograd wda solvers
deeb01 Sep 17, 2026
026866d
Merge remote-tracking branch 'origin/feat-806-wda-torch-solver' into …
deeb01 Sep 17, 2026
f1552cf
Merge branch 'master' into feat-806-wda-torch-solver
cedricvincentcuaz Sep 17, 2026
50434b9
Merge branch 'master' into feat-806-wda-torch-solver
cedricvincentcuaz Oct 3, 2026
7865c88
Merge branch 'master' into feat-806-wda-torch-solver
rflamary Oct 5, 2026
221f610
Merge branch 'master' into feat-806-wda-torch-solver
cedricvincentcuaz Oct 6, 2026
b74468d
Address review: reuse POT primitives, make wda a dispatcher
deeb01 Oct 7, 2026
b23f549
Reject single-class input in wda
deeb01 Oct 7, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions RELEASES.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,9 @@

#### New features


- `ot.dr.wda` gains `solver='torch'`, a PyTorch autodiff solver with Riemannian gradient descent, usable on installations without autograd or pymanopt, and accepting torch tensors directly (PR #858, Issue #806)
- `ot.dr` dependencies (autograd, pymanopt, scikit-learn, torch) are now imported optionally, so importing `ot.dr` no longer requires all of them; each function raises an explicit `ImportError` naming what it needs (PR #858)
- Use `ot.utils.check_marginal` (and shape-tuple support in `ot.utils.unif`) to fill and validate default marginals consistently across solvers (Gromov, low-rank, stochastic, barycenter, factored) (PR #856)
- Add stereographic spherical sliced Wasserstein distance in `ot.sliced.stereographic_sliced_wasserstein_sphere`, with its rotationally invariant extension (PR #836)
- Add Quasi-Monte Carlo sliced Wasserstein sampling (QSW/RQSW) via generalized
Expand Down
Loading
Loading