pytorch

d710c95c - Implement forward AD for scatter_reduce (#85000)

Implement forward AD for scatter_reduce (#85000) I left the case `reduction="prod"` for future work as it's a bit of a pain. Pull Request resolved: https://github.com/pytorch/pytorch/pull/85000 Approved by: https://github.com/soulitzer