John St. John c30d9b9fae Update call to `amp.autocast` from `fast_dtype` to `dtype` (#9211) ... Co-authored-by: SeanNaren <sean@grid.ai> Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>		2021-09-04 02:59:11 +00:00
..
__init__.py	FSDP with full state dict (#7487)	2021-05-24 08:11:45 +01:00
apex_amp.py	Replace `yapf` with `black` (#7783)	2021-07-26 13:37:35 +02:00
deepspeed_precision.py	Replace `yapf` with `black` (#7783)	2021-07-26 13:37:35 +02:00
double.py	Replace `yapf` with `black` (#7783)	2021-07-26 13:37:35 +02:00
fully_sharded_native_amp.py	FSDP with full state dict (#7487)	2021-05-24 08:11:45 +01:00
ipu_precision.py	Replace `yapf` with `black` (#7783)	2021-07-26 13:37:35 +02:00
mixed.py	Replace `yapf` with `black` (#7783)	2021-07-26 13:37:35 +02:00
native_amp.py	Update call to `amp.autocast` from `fast_dtype` to `dtype` (#9211)	2021-09-04 02:59:11 +00:00
precision_plugin.py	Remove `BasePlugin` (#9066)	2021-08-25 19:10:28 +00:00
sharded_native_amp.py	Add support for CPU AMP autocast (#9084)	2021-08-25 12:18:00 +00:00
tpu_bfloat.py	Replace `yapf` with `black` (#7783)	2021-07-26 13:37:35 +02:00