pytorch

bb67a287 - [DTensor] Enable Adamax foreach optimizer (#119850)

[DTensor] Enable Adamax foreach optimizer (#119850) Enable Adamax foreach optimizer and add DTensor unit test for Adamax. Pull Request resolved: https://github.com/pytorch/pytorch/pull/119850 Approved by: https://github.com/wanchaol