pytorch

07800c52 - [optim][adam] group tensors in foreach to maximize perf (#92349)

[optim][adam] group tensors in foreach to maximize perf (#92349) same idea as https://github.com/pytorch/pytorch/pull/92338 Pull Request resolved: https://github.com/pytorch/pytorch/pull/92349 Approved by: https://github.com/albanD