Rohan Varma 2ec7cd2db2 [CheckpointWrapper] Test for kwarg propagation, remove checkpoint_fn_arg support (#102679) ... Closes https://github.com/pytorch/pytorch/issues/100576 Differential Revision: [D46342398](https://our.internmc.facebook.com/intern/diff/D46342398/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/102679 Approved by: https://github.com/awgu		2023-07-28 21:18:35 +00:00
..
_checkpoint	[CheckpointWrapper] Test for kwarg propagation, remove checkpoint_fn_arg support (#102679)	2023-07-28 21:18:35 +00:00
_comm_hooks	[FSDP] Optimize away intermediate div_ for HSDP (#106034)	2023-07-28 18:36:26 +00:00
_optimizer_overlap	make fsdp folder to be public (#72084)	2022-02-02 15:50:14 +00:00
_quantization	Change docstring type callable to Callable for consistency (#82487)	2022-08-01 17:26:09 +00:00
ddp_comm_hooks	Enable fused optimizer for DP (#98270)	2023-04-13 20:16:32 +00:00
model_averaging	Convert logging f-strings to use % format, part four (#98705)	2023-04-11 13:17:59 +00:00
__init__.py
join.py	[BE] Enable ruff's UP rules and autoformat distributed/ (#105433)	2023-07-19 14:27:11 +00:00