fix: keep requires_grad when re-wrapping a lifted parameter (#4480)