Support factory kwargs in torch.nn modules (#54508)
Summary:
Continuation of https://github.com/pytorch/pytorch/pull/53144
Pull Request resolved: https://github.com/pytorch/pytorch/pull/54508
Reviewed By: mrshenli
Differential Revision: D27600457
Pulled By: jbschlosser
fbshipit-source-id: b58bfee61c3917524b4622f63ef216c27a588eb1
