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.
 
 
 
 
 
YuliangLiu0306 29386a54e6
[DTensor] refactor CommSpec (#3034)
2 years ago
..
_C
amp
auto_parallel
autochunk
builder
cli
communication
context
device
engine [doc] add deepspeed citation and copyright (#2996) 2 years ago
fx [pipeline] Add Simplified Alpa DP Partition (#2507) 2 years ago
gemini [doc] add deepspeed citation and copyright (#2996) 2 years ago
kernel [doc] add deepspeed citation and copyright (#2996) 2 years ago
logging
nn [format] applied code formatting on changed files in pull request 2997 (#3008) 2 years ago
pipeline [pipeline] Add Simplified Alpa DP Partition (#2507) 2 years ago
registry
tensor [DTensor] refactor CommSpec (#3034) 2 years ago
testing
trainer
utils [doc] add deepspeed citation and copyright (#2996) 2 years ago
zero [doc] add deepspeed citation and copyright (#2996) 2 years ago
__init__.py
constants.py
core.py
global_variables.py
initialize.py