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 1a229045af Add interface for colo tesnor dp size (#3227)		2 years ago
..
_C	…
_analyzer	[FX] refactor experimental tracer and adapt it with hf models (#3157)	2 years ago
amp	…
auto_parallel	[auto-parallel] add auto-offload feature (#3154)	2 years ago
autochunk	…
booster	[booster] added the plugin base and torch ddp plugin (#3180)	2 years ago
builder	…
checkpoint_io	[api] implemented the checkpoint io module (#3205)	2 years ago
cli	…
cluster	[API] implement device mesh manager (#3221)	2 years ago
communication	…
context	…
device	…
engine	…
fx	…
gemini	Add interface for colo tesnor dp size (#3227)	2 years ago
kernel	…
logging	…
nn	…
pipeline	…
registry	…
tensor	Add interface for colo tesnor dp size (#3227)	2 years ago
testing	…
trainer	…
utils	[lazyinit] combine lazy tensor with dtensor (#3204)	2 years ago
zero	…
__init__.py	…
constants.py	…
core.py	…
global_variables.py	…
initialize.py	…