"""
LiquidFlow: A Lightweight Image Generator combining
Liquid Neural Networks (CfC) + Mamba-2 SSD + Physics-Informed Regularization

Based on:
- CfC: "Closed-form Continuous-time Neural Networks" (Hasani et al., Nature MI 2022)
- Mamba-2: "Transformers are SSMs" (Dao & Gu, 2024)
- PINN Diffusion: "Physics-Informed Diffusion Models" (Bastek & Sun, ICLR 2025)
- DiMSUM: "Diffusion Mamba" (NeurIPS 2024)
"""

from .cfc_cell import CfCCell, CfCBlock
from .mamba2_ssd import Mamba2SSD, Mamba2Block
from .liquid_flow_block import LiquidFlowBlock, LiquidFlowStage
from .generator import LiquidFlowGenerator
from .vae_wrapper import TAESDWrapper, SDVAEWrapper
from .physics_loss import PhysicsRegularizer
from .trainer import LiquidFlowTrainer