Making large AI models cheaper, faster and more accessible
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
YH 7b13f7db18
[zero] trivial zero optimizer refactoring (#2869)
2 years ago
..
bookkeeping [zero] trivial zero optimizer refactoring (#2869) 2 years ago
__init__.py [zero] migrate zero1&2 (#1878) 2 years ago
_utils.py [zero] fix wrong import (#2777) 2 years ago
low_level_optim.py [zero] trivial zero optimizer refactoring (#2869) 2 years ago
sharded_optim_v2.py fix move fp32 shards (#1604) 2 years ago