Make `torch.lu` differentiable for wide/tall inputs + jit (#61564)
Summary:
As per title.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/61564
Reviewed By: astaff
Differential Revision: D30338136
Pulled By: mruberry
fbshipit-source-id: f01436fc90980544cdfa270feee16bb3dda21b93