Back out "Revert D26753571: [pytorch][PR] add submodules to sys.modules so their attributes can be pickled" (#53127)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/53127
Original commit changeset: cc9cc4f508af
ghstack-source-id: 122871468
Test Plan: run flake8 on the files locally
Reviewed By: malfet, janeyx99
Differential Revision: D26757859
fbshipit-source-id: 7e7bde5c1f2b434442079656e2186b500d53fdc2