distributed/_composable/__init__.py

from .checkpoint_activation import checkpoint
from .contract import _get_registry, contract
from .fully_shard import fully_shard
from .replicate import replicate