[optim] ASGD fold state updates into functional and pass list of vars rather than states (#71335)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/71335
Test Plan: Imported from OSS
Reviewed By: anjali411
Differential Revision: D33767871
Pulled By: mikaylagawarecki
fbshipit-source-id: 84ebe1fafb1c27572f08c8c8026c882dd7e054c1
(cherry picked from commit 7613ebb3914b257637ed67b1945b2a4694f4a258)
