# -*- coding: utf-8 -*-
"""
POT GPU - PyTorch GPU-accelerated Optimal Transport algorithms
Pure PyTorch implementation reproducing POT's exact behavior.
"""



from .fused_gromov_sinkhorn import (
    fused_gromov_wasserstein_sinkhorn,
    fused_gromov_wasserstein2_sinkhorn,
)

__version__ = "2.0.0"
__all__ = [
    "fused_gromov_wasserstein_sinkhorn",
    "fused_gromov_wasserstein2_sinkhorn",
]
