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: malfet
Differential Revision: D27909732
Pulled By: jbschlosser
fbshipit-source-id: d8684b2403ab7eb336371d118799146a2520bd76
