File size: 295 Bytes
917fb2a | 1 2 3 4 5 6 7 8 9 10 11 | """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"]
|