ytxiong d67be17f96 refactor(*): refactor the code with no-apex (#170) ... * support no-apex * add default for use_apex * fix lint * modify the RMSNormTorch * remove some comments * remove use_apex parameter * remove some unnecessary code * optimize the code including import * remove the import RMSNorm * remove warnings		2023-08-03 11:24:12 +08:00
..
__init__.py	initial commit	2023-07-06 12:55:23 +08:00
hybrid_zero_optim.py	feat(core/scheduler): support pipeline parallel (#98)	2023-07-24 20:52:09 +08:00
store.py	initial commit	2023-07-06 12:55:23 +08:00
utils.py	refactor(*): refactor the code with no-apex (#170)	2023-08-03 11:24:12 +08:00