[PyTorch] +40% CUDA +160% MPS Throughput Improvement on PyTorch Adam Optimizer

Part of

Cross-Architecture Performance Modeling Trends