src/matilda/__init__.py

from .config import ModelConfig, DEV_TINY, BASE_124M
from .model import Transformer

__all__ = ["ModelConfig", "DEV_TINY", "BASE_124M", "Transformer"]