magialiao 5654e63398 Use batched operations for PowerSGD ... This implements method proposed in #74907 Pull Request resolved: https://github.com/pytorch/pytorch/pull/75157 Approved by: https://github.com/wayi1, https://github.com/rohan-varma		2022-04-18 04:34:17 +00:00
..
_checkpoint	[Checkpoint] Rename file (#72748)	2022-02-16 00:08:23 +00:00
_optimizer_overlap	make fsdp folder to be public (#72084)	2022-02-02 15:50:14 +00:00
ddp_comm_hooks	Use batched operations for PowerSGD	2022-04-18 04:34:17 +00:00
model_averaging	fix PostLocalSGDOptimizer and ModelAverager average bug	2022-04-13 11:41:27 +00:00
quantization	[BE] minor improvement to dist quantization (#67401)	2021-11-21 23:31:59 -08:00
__init__.py	Make _Join, _Joinable, _JoinHook public (#62605)	2021-08-03 12:20:11 -07:00
join.py	[Join][BE] Fix typo; remove obsolete method (#72886)	2022-02-16 15:03:09 +00:00