Add pruning-aware training in torchao.prototype.pat#3429

lisjin wants to merge 1 commit intomainpytorch/ao:mainfrom

lvj/patpytorch/ao:lvj/patCopy head branch name to clipboard