src/amoe/__init__.py

"""amoe — aleph mixture-of-experts adapters (train/attach/align/detach).

The productized adapter system from the geolip-aleph research line.
"LoRA-style" refers to the attach/detach usage pattern, not the math:
these are aleph-addressed patch heads, not low-rank matrices.
"""
from .core.adapter import AdapterSpec, RelayPatchwork, BlockWithAdapter
from .core.address import AlephAddress
from .core.dispatch import AnchorDispatch, BlockWithDispatch, set_mask
from .io.checkpoint import (AnchorCheckpoint, DispatchCheckpoint,
                            load_anchor, load_dispatch,
                            import_legacy_keys)
from .runtime.attach import attach, detach, AttachHandle
from .train.config import TrainConfig, AlignConfig
from .train.trainer import train
from .train.aligner import align
from . import laws

__version__ = "0.2.2"

# The diffusion subsystem (amoe.diffusion) is imported lazily — its verbs
# live under `import amoe.diffusion as ad`. Core is pure torch; diffusers
# is only touched by bindings/samplers/data that need it.