__init__.py

from .data_parallel import MMDataParallel
from .distributed import MMDistributedDataParallel
from .scatter_gather import scatter, scatter_kwargs

__all__ = [
    'MMDataParallel', 'MMDistributedDataParallel', 'scatter', 'scatter_kwargs'
]