magialiao 7c8c8cc248 Use batched operations for PowerSGD ... This PR is a rebased version of #75157 which fixes CI issues Pull Request resolved: https://github.com/pytorch/pytorch/pull/76041 Approved by: https://github.com/albanD, https://github.com/rohan-varma		2022-04-21 03:25:09 +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-21 03:25:09 +00:00
model_averaging	Make distributed raise ImportError when not available	2022-04-20 13:05:18 +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