pytorch

b9da79d2 - Simplify _use_grad_for_differentiable (#98706)

Simplify _use_grad_for_differentiable (#98706) This makes it so dynamo can trace through it Pull Request resolved: https://github.com/pytorch/pytorch/pull/98706 Approved by: https://github.com/janeyx99