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.

Frank Lee 8518263b80 [test] fixed the triton version for testing (#2608)		2 years ago
..
apex_amp	[test] fixed the triton version for testing (#2608)	2 years ago
naive_amp	[zero] fix gradient clipping in hybrid parallelism (#2521)	2 years ago
torch_amp	[NFC] polish colossalai/amp/torch_amp/torch_amp.py code style (#2290)	2 years ago
__init__.py	[refactor] zero directory (#724)	3 years ago
amp_type.py	Develop/experiments (#59)	3 years ago