Searching in torch/distributed/algorithms