[MRG] BAPG solvers for GW and FGW

README.md

@@ -343,3 +343,7 @@ distances between Gaussian distributions](https://hal.science/hal-03197398v2/fil
 [61] Charlier, B., Feydy, J., Glaunes, J. A., Collin, F. D., & Durif, G. (2021). [Kernel operations on the gpu, with autodiff, without memory overflows](https://www.jmlr.org/papers/volume22/20-275/20-275.pdf). The Journal of Machine Learning Research, 22(1), 3457-3462.
 
 [62] H. Van Assel, C. Vincent-Cuaz, T. Vayer, R. Flamary, N. Courty (2023). [Interpolating between Clustering and Dimensionality Reduction with Gromov-Wasserstein](https://arxiv.org/pdf/2310.03398.pdf). NeurIPS 2023 Workshop Optimal Transport and Machine Learning.
+
+[63] Li, J., Tang, J., Kong, L., Liu, H., Li, J., So, A. M. C., & Blanchet, J. (2022). [A Convergent Single-Loop Algorithm for Relaxation of Gromov-Wasserstein in Graph Data](https://openreview.net/pdf?id=0jxPyVWmiiF). In The Eleventh International Conference on Learning Representations.
+
+[64] Ma, X., Chu, X., Wang, Y., Lin, Y., Zhao, J., Ma, L., & Zhu, W. (2023). [Fused Gromov-Wasserstein Graph Mixup for Graph-level Classifications](https://openreview.net/pdf?id=uqkUguNu40). In Thirty-seventh Conference on Neural Information Processing Systems.

RELEASES.md

@@ -20,6 +20,7 @@
 + Wrapper for `geomloss`` solver on empirical samples (PR #571)
 + Add `stop_criterion` feature to (un)regularized (f)gw barycenter solvers (PR #578)
 + Add `fixed_structure` and `fixed_features` to entropic fgw barycenter solver (PR #578)
++ Add new entropic BAPG solvers for GW and FGW (PR #581)
 
 #### Closed issues
 - Fix line search evaluating cost outside of the interpolation range (Issue #502, PR #504)

ot/gromov/__init__.py

@@ -20,9 +20,13 @@
 
 from ._bregman import (entropic_gromov_wasserstein,
                        entropic_gromov_wasserstein2,
+                       entropic_BAPG_gromov_wasserstein,
+                       entropic_BAPG_gromov_wasserstein2,
                        entropic_gromov_barycenters,
                        entropic_fused_gromov_wasserstein,
                        entropic_fused_gromov_wasserstein2,
+                       entropic_BAPG_fused_gromov_wasserstein,
+                       entropic_BAPG_fused_gromov_wasserstein2,
                        entropic_fused_gromov_barycenters)
 
 from ._estimators import (GW_distance_estimation, pointwise_gromov_wasserstein,
@@ -49,8 +53,10 @@
            'gromov_wasserstein', 'gromov_wasserstein2', 'fused_gromov_wasserstein',
            'fused_gromov_wasserstein2', 'solve_gromov_linesearch', 'gromov_barycenters',
            'fgw_barycenters', 'entropic_gromov_wasserstein', 'entropic_gromov_wasserstein2',
+           'entropic_BAPG_gromov_wasserstein', 'entropic_BAPG_gromov_wasserstein2',
            'entropic_gromov_barycenters', 'entropic_fused_gromov_wasserstein',
-           'entropic_fused_gromov_wasserstein2', 'entropic_fused_gromov_barycenters',
+           'entropic_fused_gromov_wasserstein2', 'entropic_BAPG_fused_gromov_wasserstein',
+           'entropic_BAPG_fused_gromov_wasserstein2', 'entropic_fused_gromov_barycenters',
            'GW_distance_estimation', 'pointwise_gromov_wasserstein', 'sampled_gromov_wasserstein',
            'semirelaxed_gromov_wasserstein', 'semirelaxed_gromov_wasserstein2',
            'semirelaxed_fused_gromov_wasserstein', 'semirelaxed_fused_gromov_wasserstein2',