pytorch

e2433e42 - [optim][adamax] group tensors in foreach to maximize perf (#92363)

[optim][adamax] group tensors in foreach to maximize perf (#92363) make foreach faster Pull Request resolved: https://github.com/pytorch/pytorch/pull/92363 Approved by: https://github.com/albanD