Back out "Revert D30745961: [DDP] Remove self.modules_params" (#64777)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/64777
Original commit changeset: 59f7cc50d369
ghstack-source-id: 138326909
Test Plan: ci
Reviewed By: H-Huang
Differential Revision: D30849442
fbshipit-source-id: bb87ba83935374d8a3ebbc29365df1417dd4f26f