training/__init__.py

"""Training-side rollout and GRPO integration helpers."""

from .grpo_trainer import (
    DEFAULT_MODEL_NAME,
    OnlineGRPOLoop,
    build_grpo_batch,
    build_reward_function,
    create_trainer,
    generate_deepspeed_config,
    load_model_and_tokenizer,
    run_episode,
)
from .curriculum import CurriculumScheduler, CurriculumStage
from .data_generator import build_prompt_dataset_from_idoft

__all__ = [
    "DEFAULT_MODEL_NAME",
    "OnlineGRPOLoop",
    "build_reward_function",
    "create_trainer",
    "load_model_and_tokenizer",
    "run_episode",
    "build_grpo_batch",
    "generate_deepspeed_config",
    "CurriculumScheduler",
    "CurriculumStage",
    "build_prompt_dataset_from_idoft",
]