Self-distillation with an EMA teacher at scale: ZeRO-3 + PEFT [P]

Self-distillation methods usually rely on an Exponential Moving Average (EMA) teachers to get synced and stable critics. However, this doesn't scale well for large models since holding both teacher and student (trained)…