DeepSpeed
9685eb92
- Syncing FusedAdam with new Apex features (#3434)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
Syncing FusedAdam with new Apex features (#3434) * Updating fused adam with new Apex bf16 support. * Removing capturable and master weight configs. * resolving pr comments --------- Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
References
#3434 - Syncing FusedAdam with new Apex features
Author
jomayeri
Parents
4716b0f7
Loading