Make scaling type configurable for MoE training#2642

danielvegamyhre merged 1 commit intomainpytorch/ao:mainfrom

danielvegamyhre/stack/26pytorch/ao:danielvegamyhre/stack/26Copy head branch name to clipboard