pytorch/torch/distributed/algorithms/model_averaging
..
__init__.py
averagers.py
hierarchical_model_averager.py
utils.py