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.

Hongxin Liu ae02d4e4f7 [bf16] add bf16 support (#3882) ... * [bf16] add bf16 support for fused adam (#3844) * [bf16] fused adam kernel support bf16 * [test] update fused adam kernel test * [test] update fused adam test * [bf16] cpu adam and hybrid adam optimizers support bf16 (#3860) * [bf16] implement mixed precision mixin and add bf16 support for low level zero (#3869) * [bf16] add mixed precision mixin * [bf16] low level zero optim support bf16 * [text] update low level zero test * [text] fix low level zero grad acc test * [bf16] add bf16 support for gemini (#3872) * [bf16] gemini support bf16 * [test] update gemini bf16 test * [doc] update gemini docstring * [bf16] add bf16 support for plugins (#3877) * [bf16] add bf16 support for legacy zero (#3879) * [zero] init context support bf16 * [zero] legacy zero support bf16 * [test] add zero bf16 test * [doc] add bf16 related docstring for legacy zero		1 year ago
..
_C	…
_analyzer	…
amp	[bf16] add bf16 support (#3882)	1 year ago
auto_parallel	…
autochunk	…
booster	[bf16] add bf16 support (#3882)	1 year ago
builder	…
checkpoint_io	…
cli	Modify torch version requirement to adapt torch 2.0 (#3896)	1 year ago
cluster	…
communication	…
context	…
device	…
engine	…
fx	…
interface	…
kernel	[bf16] add bf16 support (#3882)	1 year ago
lazy	[lazy] refactor lazy init (#3891)	1 year ago
logging	…
nn	[bf16] add bf16 support (#3882)	1 year ago
pipeline	…
registry	…
tensor	…
testing	…
trainer	…
utils	[lazy] refactor lazy init (#3891)	1 year ago
zero	[bf16] add bf16 support (#3882)	1 year ago
__init__.py	…
constants.py	…
core.py	…
global_variables.py	…
initialize.py	…