"""Model architecture for SAGE."""