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 517b63939a [autoparallel] add unary element wise handler v2 (#1674)		2 years ago
..
components_to_test	[NFC] polish test component gpt code style (#1567)	2 years ago
test_amp	…
test_auto_parallel	[autoparallel] add unary element wise handler v2 (#1674)	2 years ago
test_comm	…
test_config	…
test_context	…
test_data	…
test_data_pipeline_tensor_parallel	…
test_ddp	[feature] A new ZeRO implementation (#1644)	2 years ago
test_device	…
test_engine	…
test_fx	[autoparallel] add rotor C version (#1658)	2 years ago
test_gemini	[feature] A new ZeRO implementation (#1644)	2 years ago
test_layers	[embedding] rollback for better FAW performance (#1625)	2 years ago
test_moe	[moe] initialize MoE groups by ProcessGroup (#1640)	2 years ago
test_ops	…
test_optimizer	…
test_pipeline	[pipeline/pytree] add pytree to process args and kwargs | provide `data_process_func` to process args and kwargs after forward (#1642)	2 years ago
test_tensor	[feature] A new ZeRO implementation (#1644)	2 years ago
test_trainer	…
test_utils	…
test_zero	[feature] A new ZeRO implementation (#1644)	2 years ago
__init__.py	…