xla
138a70f4 - Add syncfree Adam and AdamW optimizers (#3294)

Commit
4 years ago
Add syncfree Adam and AdamW optimizers (#3294) * Initial commit of syncfree Adam optimizer * Initial commit of syncfree AdamW optimizer * Multiple improvements of syncfree optimizers * Fix CI script * Do not run imagenet test case Co-authored-by: Ricky Das <dasritwi@amazon.com>
Author
Parents
Loading