ColossalAI/colossalai/context/__init__.py

6 lines
193 B
Python

from .config import Config, ConfigException
from .parallel_context import ParallelContext
from .parallel_mode import ParallelMode
from .process_group_initializer import *
from .random import *