__init__.py

"""
HyperMambaLM - v1.0.0
Simple language model
"""

# Import main model
from .modeling_hypermamba import (
    HyperMambaConfig,
    HyperMambaLM,
    MetaLearningModule,
    NeuroSymbolicLayer,
    UltraMambaBlock
)

# Import utilities
from .modeling_utils import (
    AdvancedBPETokenizer,
    ModelProfiler,
    FewShotDataLoader,
    VisualizationUtils
)

__version__ = "1.0.0"
__author__ = "Nguyen Hai"

__all__ = [
    "HyperMambaConfig",
    "HyperMambaLM",
    "MetaLearningModule",
    "NeuroSymbolicLayer", 
    "UltraMambaBlock",
    "AdvancedBPETokenizer",
    "ModelProfiler",
    "FewShotDataLoader",
    "VisualizationUtils",
]

# Register models
def register_models():
    try:
        from transformers import AutoConfig, AutoModel, AutoModelForCausalLM
        
        AutoConfig.register("hypermamba", HyperMambaConfig)
        AutoModel.register(HyperMambaConfig, HyperMambaLM)
        AutoModelForCausalLM.register(HyperMambaConfig, HyperMambaLM)
        
        print("Models registered successfully!")
    except ImportError:
        print("Transformers library not found")

register_models()