super-gradients/src/super_gradients/__init__.py

from super_gradients.training import ARCHITECTURES, losses, utils, datasets_utils, DataAugmentation, SgModel, KDModel, \
    Trainer, KDTrainer
from super_gradients.common import init_trainer, is_distributed
from super_gradients.examples.train_from_recipe_example import train_from_recipe
from super_gradients.examples.train_from_kd_recipe_example import train_from_kd_recipe
from super_gradients.sanity_check import env_sanity_check

__all__ = ['ARCHITECTURES', 'losses', 'utils', 'datasets_utils', 'DataAugmentation',
           'Trainer', 'KDTrainer',
           'init_trainer', 'is_distributed', 'train_from_recipe', 'train_from_kd_recipe',
           'env_sanity_check', 'KDModel', 'SgModel']

env_sanity_check()