src/sgjm/training/__init__.py

from sgjm.training.config import (
    LossWeights,
    ModelConfig,
    OptimConfig,
    TrainingConfig,
)
from sgjm.training.backends import (
    Backend,
    ResolvedBackend,
    detect_backend,
    resolve_backend,
)

__all__ = [
    "Backend",
    "LossWeights",
    "ModelConfig",
    "OptimConfig",
    "ResolvedBackend",
    "TrainingConfig",
    "detect_backend",
    "resolve_backend",
]