Jiarui Fang d209aff684 Add FreqAwareEmbeddingBag (#1421)		2022-08-09 16:26:12 +08:00
..
components_to_test	…
test_amp	…
test_comm	[communication] add p2p_v2.py to support communication with List[Any] (#1407)	2022-08-09 11:40:04 +08:00
test_config	…
test_context	…
test_data	…
test_data_pipeline_tensor_parallel	…
test_ddp	[zero] alleviate memory usage in ZeRODDP state_dict (#1398)	2022-08-02 15:49:13 +08:00
test_device	[device] add DeviceMesh class to support logical device layout (#1394)	2022-08-02 19:23:48 +08:00
test_engine	…
test_fx	…
test_gemini	…
test_layers	…
test_moe	…
test_optimizer	[hotfix] fix CPUAdam kernel nullptr (#1410)	2022-08-05 19:45:45 +08:00
test_pipeline	…
test_tensor	Add FreqAwareEmbeddingBag (#1421)	2022-08-09 16:26:12 +08:00
test_trainer	…
test_utils	…
test_zero	…
__init__.py	…