lightning/pytorch_lightning/callbacks/__init__.py

from .pt_callbacks import EarlyStopping, ModelCheckpoint, GradientAccumulationScheduler

__all__ = [
    'EarlyStopping',
    'ModelCheckpoint',
    'GradientAccumulationScheduler',
]