pytorch

3e9df622 - [mta] implement `_foreach_pow` (#92303)

[mta] implement `_foreach_pow` (#92303) Mainly for foreach path of `Adam` and `AdamW` rel: https://github.com/pytorch/pytorch/issues/58833 Pull Request resolved: https://github.com/pytorch/pytorch/pull/92303 Approved by: https://github.com/albanD