"""Training stack for SAGE."""