Frank Lee ed19290560 [booster] implemented mixed precision class (#3151) ... * [booster] implemented mixed precision class * polish code		2023-03-17 11:00:15 +08:00
..
components_to_test	…
kit	[booster] implemented mixed precision class (#3151)	2023-03-17 11:00:15 +08:00
test_amp	…
test_analyzer	[analyzer] a minimal implementation of static graph analyzer (#2852)	2023-03-10 13:21:05 +08:00
test_auto_parallel	[hotfix] skip auto checkpointing tests (#3029)	2023-03-07 15:50:00 +08:00
test_autochunk	[autochunk] support complete benchmark (#3121)	2023-03-13 17:42:37 +08:00
test_booster/test_mixed_precision	[booster] implemented mixed precision class (#3151)	2023-03-17 11:00:15 +08:00
test_comm	…
test_config	…
test_context	…
test_data	…
test_data_pipeline_tensor_parallel	…
test_ddp	…
test_device	…
test_engine	…
test_fx	[test] add torchrec models to test model zoo (#3139)	2023-03-15 05:46:04 +00:00
test_gemini	…
test_layers	…
test_moe	…
test_ops	…
test_optimizer	…
test_pipeline	…
test_tensor	[DTensor] refactor dtensor with new components (#3089)	2023-03-14 16:25:47 +08:00
test_trainer	…
test_utils	…
test_zero	…
__init__.py	…