"""needle_torch — PyTorch port of the Needle Simple Attention Network.

Public API used by Tasks 2C, 2D, 7, 8:
    from needle_torch import NeedleModel, TransformerConfig
"""

from .config import TransformerConfig
from .model import NeedleModel

__all__ = ["NeedleModel", "TransformerConfig"]