Frank Lee e2089c5c15 adapted for sequence parallel (#163)		2022-01-20 13:44:51 +08:00
..
apex_amp	AMP docstring/markdown update (#160)	2022-01-18 18:33:36 +08:00
naive_amp	adapted for sequence parallel (#163)	2022-01-20 13:44:51 +08:00
torch_amp	AMP docstring/markdown update (#160)	2022-01-18 18:33:36 +08:00
__init__.py	update examples and sphnix docs for the new api (#63)	2021-12-13 22:07:01 +08:00
amp_type.py	Develop/experiments (#59)	2021-12-09 15:08:29 +08:00