"""
MiniMind Max2 Model Package
A lightweight, efficient language model designed for edge deployment.
"""

from .mind2_model import (
    Max2ForCausalLM,
    Max2Model,
    Mind2ForCausalLM,
    Mind2Model,
    create_model
)
from .components import (
    Max2Attention,
    Max2MoE,
    Max2DecoderLayer,
    Max2RMSNorm,
    Max2RotaryEmbedding,
    Max2MLP,
    Max2Expert,
    # Backward compatibility
    Mind2Attention,
    Mind2MoE,
    Mind2DecoderLayer,
    Mind2RMSNorm,
    Mind2RotaryEmbedding,
)

__all__ = [
    # Max2 (primary)
    "Max2ForCausalLM",
    "Max2Model",
    "Max2Attention",
    "Max2MoE",
    "Max2DecoderLayer",
    "Max2RMSNorm",
    "Max2RotaryEmbedding",
    "Max2MLP",
    "Max2Expert",
    # Mind2 (backward compatibility)
    "Mind2ForCausalLM",
    "Mind2Model",
    "Mind2Attention",
    "Mind2MoE",
    "Mind2DecoderLayer",
    "Mind2RMSNorm",
    "Mind2RotaryEmbedding",
    # Factory
    "create_model",
]

__version__ = "1.0.0"