Will Constable c82c46ccc7 [C10D] support group_src/dst in broadcast/reduce ops (#140843) ... Also add mypy annotations Partially addresses RFC 0042 (pytorch/rfcs#71) See more details/motivation in #140460 Pull Request resolved: https://github.com/pytorch/pytorch/pull/140843 Approved by: https://github.com/kwen2501		2024-11-19 01:23:08 +00:00
..
_checkpoint	[BE] [Reland] Make nn.Module state_dict load_state_dict pre-hook and state_dict post-hook public (#131690)	2024-07-26 18:14:07 +00:00
_comm_hooks	[BE]: Update mypy to 1.11.2 (#133816)	2024-09-16 19:44:11 +00:00
_optimizer_overlap	[BE][Easy] enable UFMT for torch/distributed/{algorithms,autograd,benchmarks,checkpoint,elastic}/ (#128866)	2024-06-18 13:51:53 +00:00
_quantization	[BE][Easy] enable UFMT for torch/distributed/{algorithms,autograd,benchmarks,checkpoint,elastic}/ (#128866)	2024-06-18 13:51:53 +00:00
ddp_comm_hooks	[C10D] support group_src/dst in broadcast/reduce ops (#140843)	2024-11-19 01:23:08 +00:00
model_averaging	Use device-agnostic runtime API in distributed DDP/FSDP instead of cuda device specific. (#137678)	2024-11-13 05:32:19 +00:00
__init__.py	[BE][Easy] enable UFMT for torch/distributed/{algorithms,autograd,benchmarks,checkpoint,elastic}/ (#128866)	2024-06-18 13:51:53 +00:00
join.py	[BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200)	2024-08-15 15:50:19 +00:00