Sean Naren d01e8fdc86 [fix] Use training type plugin hook when saving (FSDP 1/n) (#6321) ... * Rely on training type plugin when saving * Add better typing to training type plugin		2021-03-04 18:09:33 +00:00
..
__init__.py	PoC: Accelerator refactor (#5743)	2021-02-12 15:48:56 -05:00
accelerator.py	[fix] Use training type plugin hook when saving (FSDP 1/n) (#6321)	2021-03-04 18:09:33 +00:00
cpu.py	type accelerators (#6148)	2021-02-25 06:42:23 +00:00
gpu.py	type accelerators (#6148)	2021-02-25 06:42:23 +00:00
tpu.py	[bugfix] TPU + all_gather + SingleTPU shouldn't call xm.all_gather (#6296)	2021-03-03 08:54:20 -05:00