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: bdhirsh
Differential Revision: D27855386
Pulled By: jbschlosser
fbshipit-source-id: dabd505d2a04208e74b158570fb2859c736eea2c