ColossalAI/colossalai/context/random/__init__.py

from ._helper import (
    add_seed,
    get_current_mode,
    get_seeds,
    get_states,
    moe_set_seed,
    reset_seeds,
    seed,
    set_mode,
    set_seed_states,
    sync_states,
    with_seed,
)

__all__ = [
    'seed', 'set_mode', 'with_seed', 'add_seed', 'get_seeds', 'get_states', 'get_current_mode', 'set_seed_states',
    'sync_states', 'moe_set_seed', 'reset_seeds'
]