Kaushik B a72a7992a2 Update clip gradients signature for precision plugins (#6764)		2021-03-31 17:06:48 +05:30
..
__init__.py	Feature/double precision (#6595)	2021-03-24 15:47:58 +05:30
apex_amp.py	[bug] Update broadcast + reduce decision ModelCheckpoint] (#6410)	2021-03-14 17:14:27 +00:00
deepspeed_precision.py	Update clip gradients signature for precision plugins (#6764)	2021-03-31 17:06:48 +05:30
double.py	DeepSpeed ZeRO Update (#6546)	2021-03-30 13:39:02 -04:00
mixed.py	Add mypy typing to precision plugins. (#6149)	2021-02-26 14:27:16 +01:00
native_amp.py	Add AMP for validation, prediction and testing (#6565)	2021-03-20 23:15:49 +00:00
precision_plugin.py	Update clip gradients signature for precision plugins (#6764)	2021-03-31 17:06:48 +05:30
sharded_native_amp.py	Update clip gradients signature for precision plugins (#6764)	2021-03-31 17:06:48 +05:30
tpu_bfloat.py	Add mypy typing to precision plugins. (#6149)	2021-02-26 14:27:16 +01:00