Rohan Varma 44fe851feb [WIP] Fix non-reentrant hooks based checkpointing ... Pull Request resolved: https://github.com/pytorch/pytorch/pull/78752 Approved by: https://github.com/albanD		2022-06-14 01:13:33 +00:00
..
_checkpoint	[WIP] Fix non-reentrant hooks based checkpointing	2022-06-14 01:13:33 +00:00
_optimizer_overlap	make fsdp folder to be public (#72084)	2022-02-02 15:50:14 +00:00
ddp_comm_hooks	[Model Averaging] Support disabling post-local gradient sync (#76723)	2022-05-16 18:09:09 +00:00
model_averaging	[Model Averaging] Make an error message more clear in hierarchical_model_averager.py	2022-04-26 15:20:51 +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