pytorch

b193f295 - Add capturable ASGD impl (#107857)

Add capturable ASGD impl (#107857) Add capturable ASGD impl + test Pull Request resolved: https://github.com/pytorch/pytorch/pull/107857 Approved by: https://github.com/janeyx99