SIGN IN SIGN UP
hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

0 0 21 Python

[Feature] auto-cast optimizers to distributed version (#5746)

* auto-cast optimizers to distributed

* fix galore casting

* logger

---------

Co-authored-by: Edenzzzz <wtan45@wisc.edu>
E
Edenzzzz committed
5f8c0a0ac3b52a71b664c3e36dd1a8cef40f428d
Parent: 2fc85ab
Committed by GitHub <noreply@github.com> on 5/24/2024, 9:24:16 AM