"""Communication algorithms package for sagellm-comm.

Provides collective algorithm implementations (Ring / Tree / Hybrid All-Reduce)
that run as pure-Python simulations and interface with the CommBackend.
"""

from __future__ import annotations

from sagellm_comm.algorithms.allreduce import (
    AllReduceAlgorithm,
    HybridAllReduce,
    RingAllReduce,
    TreeAllReduce,
)
from sagellm_comm.algorithms.selector import AlgorithmSelector, SelectorConfig

__all__ = [
    "AllReduceAlgorithm",
    "RingAllReduce",
    "TreeAllReduce",
    "HybridAllReduce",
    "AlgorithmSelector",
    "SelectorConfig",
]
