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.

Jiarui Fang 8d653af408 add a common util for hooks registered on parameter. (#292)		3 years ago
..
amp	fixed apex import (#227)	3 years ago
builder	add pytorch hooks (#179)	3 years ago
communication	moved env variables to global variables; (#215)	3 years ago
context	moved env variables to global variables; (#215)	3 years ago
engine	add a common util for hooks registered on parameter. (#292)	3 years ago
kernel	Optimized MoE layer and fixed some bugs;	3 years ago
logging	fixed mkdir conflict and align yapf config with flake (#220)	3 years ago
nn	Added TPExpert for special situation	3 years ago
registry	add pytorch hooks (#179)	3 years ago
trainer	moved env variables to global variables; (#215)	3 years ago
utils	Feature/zero (#279)	3 years ago
zero	Feature/zero (#279)	3 years ago
__init__.py	…
constants.py	moved env variables to global variables; (#215)	3 years ago
core.py	…
global_variables.py	Optimized MoE layer and fixed some bugs;	3 years ago
initialize.py	bug fix: pass hook_list to engine (#273)	3 years ago