[MRG] split ot.bregman

RELEASES.md

@@ -23,6 +23,8 @@
 - Handle documentation and warnings when integers are provided to (f)gw solvers based on cg (Issue #530, PR #559)
 - Correct independence of `fgw_barycenters` to `init_C` and `init_X` (Issue #547, PR #566)
 - Avoid precision change when computing norm using PyTorch backend (Discussion #570, PR #572)
+- Create `ot/bregman/`repository (Issue #567, PR #569)
+
 
 ## 0.9.1
 *August 2023*

ot/bregman/__init__.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+"""
+Solvers related to Bregman projections for entropic regularized OT
+"""
+
+# Author: Remi Flamary <remi.flamary@unice.fr>
+#         Cédric Vincent-Cuaz <cedvincentcuaz@gmail.com>
+#
+# License: MIT License
+
+from ._utils import (geometricBar,
+                     geometricMean,
+                     projR,
+                     projC)
+
+from ._sinkhorn import (sinkhorn,
+                        sinkhorn2,
+                        sinkhorn_knopp,
+                        sinkhorn_log,
+                        greenkhorn,
+                        sinkhorn_stabilized,
+                        sinkhorn_epsilon_scaling)
+
+from ._barycenter import (barycenter,
+                          barycenter_sinkhorn,
+                          free_support_sinkhorn_barycenter,
+                          barycenter_stabilized,
+                          barycenter_debiased,
+                          jcpot_barycenter)
+
+from ._convolutional import (convolutional_barycenter2d,
+                             convolutional_barycenter2d_debiased)
+
+from ._empirical import (empirical_sinkhorn,
+                         empirical_sinkhorn2,
+                         empirical_sinkhorn_divergence)
+
+from ._screenkhorn import (screenkhorn)
+
+from ._dictionary import (unmix)
+
+
+__all__ = ['geometricBar', 'geometricMean', 'projR', 'projC',
+           'sinkhorn', 'sinkhorn2', 'sinkhorn_knopp', 'sinkhorn_log',
+           'greenkhorn', 'sinkhorn_stabilized', 'sinkhorn_epsilon_scaling',
+           'barycenter', 'barycenter_sinkhorn', 'free_support_sinkhorn_barycenter',
+           'barycenter_stabilized', 'barycenter_debiased', 'jcpot_barycenter',
+           'convolutional_barycenter2d', 'convolutional_barycenter2d_debiased',
+           'empirical_sinkhorn', 'empirical_sinkhorn2',
+           'empirical_sinkhorn_divergence',
+           'screenkhorn',
+           'unmix'
+           ]