Added tutorial on Sinkhorn divergence (#282)

* fixed the docs about the implicit_diff argument * Added a tutorial for the Sinkhorn divergence * Updated the list of tutorials * made requested fixes * improved myST, comments, added colab * trigger precommit * added tildes and ran linter * lint * refactored, default epsilon vs bigger one, static_b * lint * Trigger commit * changed a word * Changed openreview links
ott-jax · Mar 28, 2023 · 9e709f7 · 9e709f7
1 parent ab256de
commit 9e709f7
Show file tree

Hide file tree

Showing 4 changed files with 26,872 additions and 11 deletions.
diff --git a/docs/references.bib b/docs/references.bib
@@ -194,7 +194,7 @@ @misc{richter-powell:21
 @inproceedings{bunne:22,
   author    = {Bunne, Charlotte and Krause, Andreas and marco cuturi},
   editor    = {Oh, Alice H. and Agarwal, Alekh and Belgrave, Danielle and Cho, Kyunghyun},
-  url       = {https://openreview.net/forum?id=sPNtVVUq7wi},
+  url       = {https://arxiv.org/abs/2206.14262},
   booktitle = {Advances in Neural Information Processing Systems},
   title     = {Supervised Training of Conditional Monge Maps},
   year      = {2022},
@@ -582,7 +582,7 @@ @misc{amos:22
 
 @inproceedings{korotin:21,
   author    = {Korotin, Alexander and Egiazarian, Vage and Asadulaev, Arip and Safin, Alexander and Burnaev, Evgeny},
-  url       = {https://openreview.net/forum?id=bEoxzW_EXsa},
+  url       = {https://arxiv.org/abs/1909.13082},
   booktitle = {International Conference on Learning Representations},
   title     = {Wasserstein-2 Generative Networks},
   year      = {2021},

diff --git a/docs/tutorials/index.rst b/docs/tutorials/index.rst
@@ -18,6 +18,7 @@ Linear Optimal Transport
     notebooks/OTT_&_POT
     notebooks/Hessians
     notebooks/LRSinkhorn
+    notebooks/sinkhorn_divergence_gradient_flow
 
 Barycenters
 ^^^^^^^^^^^

diff --git a/docs/tutorials/notebooks/point_clouds.ipynb b/docs/tutorials/notebooks/point_clouds.ipynb
diff --git a/docs/tutorials/notebooks/sinkhorn_divergence_gradient_flow.ipynb b/docs/tutorials/notebooks/sinkhorn_divergence_gradient_flow.ipynb