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: albanD
Differential Revision: D27939544
Pulled By: jbschlosser
fbshipit-source-id: 4bf517e5f74f093e27ca38a85e732da65e44d805