"""
Loss functions for multi-modal multi-scale graph model.
"""

from .alignment import CLIPLoss, InfoNCELoss, ContrastiveLoss
from .reconstruction import MSELoss, L1Loss, KLDivLoss
from .task_losses import ClassificationLoss, RegressionLoss, GenerationLoss

__all__ = [
    # Alignment losses
    'CLIPLoss',
    'InfoNCELoss', 
    'ContrastiveLoss',
    
    # Reconstruction losses
    'MSELoss',
    'L1Loss',
    'KLDivLoss',
    
    # Task-specific losses
    'ClassificationLoss',
    'RegressionLoss',
    'GenerationLoss',
] 