annotate torch.autograd.* modules (#45004)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/44638
Pull Request resolved: https://github.com/pytorch/pytorch/pull/45004
Reviewed By: VitalyFedyunin
Differential Revision: D24113562
Pulled By: ezyang
fbshipit-source-id: a85018b7e08b2fe6cf2bc14a217eb418cb2b9de4
