DeepSpeed
bbb4bfcf
- make muon optimizer totally running on GPU
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
65 days ago
make muon optimizer totally running on GPU Signed-off-by: Guokai Ma <guokai.ma@intel.com>
References
#7648 - Put Muon optimizer momentum buffer on GPU
Author
delock
Committer
delock
Parents
7af561c2
Loading