torch distributed: add support for user-specified parameter synchronization - #1612

Open

NeoLegends wants to merge 7 commits into

moritz-torch-custom-reduce

Commits on Sep 4, 2024

Commits on Sep 6, 2024

chore: Simplify implementation to support only callback
NeoLegends
committed
e106103
fix text
NeoLegends
committed
4540bdd
remove torch.no_grad() for more flexibility
NeoLegends
committed
d666d33