lightning/pytorch_lightning/plugins/precision
Rohit Gupta 06b8f82b8a
Update API references in doc (#11357)
2022-01-07 15:56:17 +01:00
..
__init__.py Remove the "_precision" suffix from some precision plugin files (#10052) 2021-11-19 17:37:39 +00:00
apex_amp.py Deprecate Trainer.training_type_plugin in favor of trainer.strategy (#11141) 2021-12-22 02:11:43 +00:00
deepspeed.py Improve typing for plugins (#10742) 2021-11-26 20:14:58 +00:00
double.py Introduce `PrecisionPlugin.forward_context()` (#9988) 2021-10-18 12:58:19 +00:00
fully_sharded_native_amp.py Fix deadlinks in docs (#10739) 2021-12-02 09:05:32 +00:00
ipu.py Update API references in doc (#11357) 2022-01-07 15:56:17 +01:00
mixed.py CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00
native_amp.py Improve typing for plugins (#10742) 2021-11-26 20:14:58 +00:00
precision_plugin.py Fix BF16 teardown for TPU precision plugin (#10990) 2021-12-22 03:47:14 +00:00
sharded_native_amp.py Pass the scaler as an input to `NativeMixedPrecisionPlugin` (#10055) 2021-10-28 14:13:53 +00:00
tpu.py Update API references in doc (#11357) 2022-01-07 15:56:17 +01:00
tpu_bf16.py Fix BF16 teardown for TPU precision plugin (#10990) 2021-12-22 03:47:14 +00:00